Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:DECLARE_GPU_SPECS_INDEX_OP (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/core/kernels/
scatter_nd_op_gpu.cu.cc	155 #define DECLARE_GPU_SPECS_INDEX_OP(T, Index, op) \ 165 DECLARE_GPU_SPECS_INDEX_OP(T, Index, scatter_nd_op::UpdateOp::ASSIGN); \ 166 DECLARE_GPU_SPECS_INDEX_OP(T, Index, scatter_nd_op::UpdateOp::ADD); \ 167 DECLARE_GPU_SPECS_INDEX_OP(T, Index, scatter_nd_op::UpdateOp::SUB) 179 #undef DECLARE_GPU_SPECS_INDEX_OP
scatter_nd_op.cc	525 #define DECLARE_GPU_SPECS_INDEX_OP(T, Index, op) \ 535 DECLARE_GPU_SPECS_INDEX_OP(T, Index, scatter_nd_op::UpdateOp::ASSIGN); \ 536 DECLARE_GPU_SPECS_INDEX_OP(T, Index, scatter_nd_op::UpdateOp::ADD); \ 537 DECLARE_GPU_SPECS_INDEX_OP(T, Index, scatter_nd_op::UpdateOp::SUB) 550 #undef DECLARE_GPU_SPECS_INDEX_OP

Completed in 67 milliseconds