HomeSort by relevance Sort by last modified time
    Searched refs:DECLARE_GPU_SPECS_INDEX_OP_IXDIM (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
scatter_nd_op_gpu.cu.cc 152 #define DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, IXDIM) \
156 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 1); \
157 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 2); \
158 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 3); \
159 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 4); \
160 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 5); \
161 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 6); \
162 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 7);
scatter_nd_op.cc 514 #define DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, IXDIM) \
526 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 1); \
527 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 2); \
528 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 3); \
529 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 4); \
530 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 5); \
531 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 6); \
532 DECLARE_GPU_SPECS_INDEX_OP_IXDIM(T, Index, op, 7);

Completed in 201 milliseconds