Home
last modified time | relevance | path

Searched defs:TENSOR_TYPE (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dpack_test.cc58 TensorType TENSOR_TYPE = member
/external/ComputeLibrary/src/core/CL/cl_kernels/
Dtile_helpers.h489 #define V_LOAD(DATA_TYPE, WIDTH, TENSOR_TYPE, TENSOR, X, Y, STRIDE_Y) V_LOAD_STR(DATA_TYPE, WIDTH, … argument
490 #define V_LOAD_STR(DATA_TYPE, WIDTH, TENSOR_TYPE, TENSOR, X, Y, STRIDE_Y) V_LOAD_##TENSOR_TYPE(DATA… argument
508 #define V_STORE(DATA_TYPE, WIDTH, TENSOR_TYPE, TENSOR, X, Y, STRIDE_Y, VALUES) V_STORE_STR(DATA_TYP… argument
509 #define V_STORE_STR(DATA_TYPE, WIDTH, TENSOR_TYPE, TENSOR, X, Y, STRIDE_Y, VALUES) V_STORE_##TENSOR… argument
531 #define T_LOAD(DATA_TYPE, HEIGHT, WIDTH, TENSOR_TYPE, TENSOR, X, Y, YI_MULTIPLIER, STRIDE_Y, dst) … argument
552 #define T_LOAD_INDIRECT(DATA_TYPE, HEIGHT, WIDTH, TENSOR_TYPE, TENSOR, X, STRIDE_Y, indirect_y, dst… argument
578 #define T_LOAD_INDIRECT_WIDTH_SELECT(DATA_TYPE, HEIGHT, WIDTH0, WIDTH1, TENSOR_TYPE, TENSOR, X, STR… argument
614 #define T_LOAD_NHWC(DATA_TYPE, TILE_HEIGHT, TILE_WIDTH, TILE_CHANNELS, TENSOR_TYPE, TENSOR, B, Y, X… argument
651 #define T_LOAD_NHWC_WITH_DILATION(DATA_TYPE, TILE_HEIGHT, TILE_WIDTH, TILE_CHANNELS, TENSOR_TYPE, T… argument
697 #define T_LOAD_NHWC_INDIRECT(DATA_TYPE, TILE_AREA, TILE_CHANNELS, TENSOR_TYPE, TENSOR, B, Y, X, C, … argument
[all …]
/external/pytorch/aten/src/ATen/native/cuda/
DIndexing.cu893 #define SMALL_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, SELF_DIM, SOURCE_DIM, IDX_DIM) \ in index_add_cuda_impl() argument
901 #define LARGE_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, \ in index_add_cuda_impl() argument
1066 #define SMALL_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, SELF_DIM, SOURCE_DIM, IDX_DIM) … in index_reduce_func_cuda_impl() argument
1074 #define LARGE_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, \ in index_reduce_func_cuda_impl() argument
1360 #define SMALL_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, DST_DIM, SRC_DIM, IDX_DIM) \ in index_select_out_cuda_impl() argument
1368 #define LARGE_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, \ in index_select_out_cuda_impl() argument
/external/tensorflow/tensorflow/lite/kernels/
Dunpack_test.cc95 TensorType TENSOR_TYPE = member
Dpack_test.cc202 TensorType TENSOR_TYPE = member