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

  /external/tensorflow/tensorflow/core/kernels/
scatter_functor_gpu.cu.cc 26 #define DEFINE_GPU_SPECS_OP(T, Index, op) \
30 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ASSIGN); \
31 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ADD); \
32 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::SUB); \
33 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MUL); \
34 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::DIV);
48 #undef DEFINE_GPU_SPECS_OP
scatter_op_gpu.cu.cc 27 #define DEFINE_GPU_SPECS_OP(T, Index, op) \
31 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ASSIGN); \
32 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ADD); \
33 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::SUB); \
34 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MUL); \
35 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::DIV);
48 #undef DEFINE_GPU_SPECS_OP

Completed in 56 milliseconds