Searched defs:Tensor3DIterator (Results 1 – 2 of 2) sorted by relevance
| /external/ComputeLibrary/src/core/GLES_COMPUTE/cs_shaders/ |
| D | pooling_layer.cs | 109 float calculate_max(const int, Tensor3DIterator, const int, const int, const int, const int, const … in calculate_max() argument
110 float calculate_avg(const int, Tensor3DIterator, const int, const int, const int, const int, const … in calculate_avg() argument
455 vec2 calculate_max(const int, Tensor3DIterator, const int, const int, const int, const int, const i… in calculate_max() argument
456 vec2 calculate_avg(const int, Tensor3DIterator, const int, const int, const int, const int, const i… in calculate_avg() argument
|
| D | helpers_cs.h | 85 struct Tensor3DIterator struct
87 int current_offset_in_bytes; /**< Current offset of tensor (in bytes) */
88 int stride_x; /**< Stride of the tensor in X dimension (in bytes) */
89 int stride_y; /**< Stride of the tensor in Y dimension (in bytes) */
90 int stride_z; /**< Stride of the tensor in Z dimension (in bytes) */
91 int element_shift; /**< The number of bits to shift by for one element */
|