|
| generic | init (t, n1, n2, n3, n4, name) |
| | Initialise a tensor of size n1*n2*n3*n4.
|
| |
| procedure, pass(t) | free (t) |
| | Deallocate a tensor.
|
| |
| procedure, pass(t) | size (t) |
| | Returns the number of entries in the tensor.
|
| |
| procedure, pass(t) | copy_from (t, memdir, sync) |
| | Copy between host and device.
|
| |
| procedure, pass(t) | get_n1 (t) |
| | Returns the size of the first dimension.
|
| |
| procedure, pass(t) | get_n2 (t) |
| | Returns the size of the second dimension.
|
| |
| procedure, pass(t) | get_n3 (t) |
| | Returns the size of the third dimension.
|
| |
| procedure, pass(t) | get_n4 (t) |
| | Returns the size of the fourth dimension.
|
| |
| procedure, pass(t) | tensor4_assign_tensor4 (t, w) |
| | Assignment \( t = w \).
|
| |
| procedure, pass(t) | tensor4_assign_scalar (t, s) |
| | Assignment \( t = s \).
|
| |
| generic | assignment (t, w) |
| | Assignment \( t = w \).
|
| |
| generic | assignment (t, s) |
| | Assignment \( t = s \).
|
| |
|
| procedure, pass, private | init_dims (t, n1, n2, n3, n4, name) |
| | Initialise a tensor of size n1*n2*n3*n4.
|
| |
| procedure, pass, private | alloc (t, n1, n2, n3, n4) |
| | Allocate a tensor of size n1*n2*n3*n4.
|
| |
Definition at line 45 of file tensor4.f90.
◆ alloc()
| procedure, pass, private tensor4::tensor4_t::alloc |
( |
class(tensor4_t), intent(inout) |
t, |
|
|
integer, intent(in) |
n1, |
|
|
integer, intent(in) |
n2, |
|
|
integer, intent(in) |
n3, |
|
|
integer, intent(in) |
n4 |
|
) |
| |
|
private |
- Parameters
-
| t | Tensor to allocate. |
| n1 | Size of the first dimension. |
| n2 | Size of the second dimension. |
| n3 | Size of the third dimension. |
| n4 | Size of the fourth dimension. |
Definition at line 86 of file tensor4.f90.
◆ assignment() [1/2]
- Parameters
-
| t | Tensor to assign to. |
| s | Scalar to fill the tensor with. |
Definition at line 82 of file tensor4.f90.
◆ assignment() [2/2]
- Parameters
-
| t | Tensor to assign to. |
| w | Tensor to assign from. |
Definition at line 82 of file tensor4.f90.
◆ copy_from()
- Parameters
-
| t | Tensor to copy to/from device/host. |
| memdir | Direction to copy (HOST_TO_DEVICE or DEVICE_TO_HOST). |
| sync | Whether the memcopy is to be blocking or not. |
Definition at line 68 of file tensor4.f90.
◆ free()
◆ get_n1()
◆ get_n2()
◆ get_n3()
◆ get_n4()
◆ init()
| generic tensor4::tensor4_t::init |
( |
class(tensor4_t), intent(inout) |
t, |
|
|
integer, intent(in) |
n1, |
|
|
integer, intent(in) |
n2, |
|
|
integer, intent(in) |
n3, |
|
|
integer, intent(in) |
n4, |
|
|
character(len=*), intent(in), optional |
name |
|
) |
| |
- Note
- Declared as a generic (rather than a plain binding) so that types extending tensor4_t can override
init_dims and have that override correctly replace this specific within the inherited init generic, instead of ambiguously merging with it.
Definition at line 62 of file tensor4.f90.
◆ init_dims()
| procedure, pass, private tensor4::tensor4_t::init_dims |
( |
class(tensor4_t), intent(inout) |
t, |
|
|
integer, intent(in) |
n1, |
|
|
integer, intent(in) |
n2, |
|
|
integer, intent(in) |
n3, |
|
|
integer, intent(in) |
n4, |
|
|
character(len=*), intent(in), optional |
name |
|
) |
| |
|
private |
- Parameters
-
| t | Tensor to initialise. |
| n1 | Size of the first dimension. |
| n2 | Size of the second dimension. |
| n3 | Size of the third dimension. |
| n4 | Size of the fourth dimension. |
| name | Optional name of the tensor. |
Definition at line 56 of file tensor4.f90.
◆ size()
◆ tensor4_assign_scalar()
- Parameters
-
| t | Tensor to assign to. |
| s | Scalar to fill the tensor with. |
Definition at line 80 of file tensor4.f90.
◆ tensor4_assign_tensor4()
- Parameters
-
| t | Tensor to assign to. |
| w | Tensor to assign from. |
Definition at line 78 of file tensor4.f90.
◆ n1
◆ n2
◆ n3
◆ n4
◆ name
| character(len=neko_varname_len) tensor4::tensor4_t::name = "" |
◆ x_d
The documentation for this type was generated from the following file: