Searched defs:Tindices (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | listdiff_op.cc | 76 auto Tindices = indices->vec<Tidx>(); in Compute() local
|
| D | gather_nd_op_cpu_impl.h | 45 const Index slice_size, typename TTypes<Index>::ConstMatrix Tindices, in GatherNdSliceGenerator()
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 141 #define DEFINE(T, Tindices) \ argument
|
| D | training_ops.cc | 1295 #define REGISTER_KERNELS(T, D, Tindices) \ argument
1554 #define REGISTER_KERNELS(T, Tindices) \ argument
1958 #define REGISTER_KERNELS(D, T, Tindices) \ argument
2091 #define REGISTER_KERNELS(D, T, Tindices) \ argument
2238 #define REGISTER_KERNELS(D, T, Tindices) \ argument
2567 #define REGISTER_KERNELS(T, Tindices) \ argument
2941 #define REGISTER_KERNELS(D, T, Tindices) \ argument
3001 #define REGISTER_KERNELS(D, T, Tindices) \ argument
3261 #define REGISTER_KERNELS(T, Tindices) \ argument
3464 #define REGISTER_KERNELS(T, D, Tindices) \ argument
[all …]
|
| D | sparse_tensor_dense_matmul_op.cc | 182 #define DECLARE_GPU_SPEC(T, Tindices, ADJ_A, ADJ_B) \ argument
|