Home
last modified time | relevance | path

Searched defs:IXDIM (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgather_nd_op.h133 #define PARAMS_CASE(IXDIM) \ in DoGatherNd() argument
Dscatter_nd_op_gpu.cu.cc168 #define DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, IXDIM) \ argument
Dscatter_nd_op.cc942 #define PARAMS_CASE(IXDIM) \ in DoScatterNd() argument
986 #define DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, IXDIM) \ argument