/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 30 template <typename T, typename Tindices, bool ADJ_A, bool ADJ_B> 33 const Tindices* a_indices, in SparseTensorDenseMatMulKernel() 65 template <typename T, typename Tindices, bool ADJ_A, bool ADJ_B> 66 struct SparseTensorDenseMatMulFunctor<GPUDevice, T, Tindices, ADJ_A, ADJ_B> { 69 typename TTypes<Tindices>::ConstMatrix a_indices, in Compute() 84 SparseTensorDenseMatMulKernel<T, Tindices, ADJ_A, ADJ_B> in Compute() 95 #define DEFINE(T, Tindices) \ argument 97 GPUDevice, T, Tindices, false, false>; \ 99 GPUDevice, T, Tindices, false, true>; \ 101 GPUDevice, T, Tindices, true, false>; \ [all …]
|
D | sparse_tensor_dense_matmul_op.cc | 32 template <typename Device, typename T, typename Tindices> 138 Device, T, Tindices, ADJ_A, \ in Compute() 140 a_indices->matrix<Tindices>(), a_values->vec<T>(), \ in Compute() 180 #define DECLARE_GPU_SPEC(T, Tindices, ADJ_A, ADJ_B) \ argument 183 GPUDevice, T, Tindices, ADJ_A, \ 185 TTypes<Tindices>::ConstMatrix a_indices, \ 189 GPUDevice, T, Tindices, ADJ_A, ADJ_B>; 242 template <typename T, typename Tindices, bool ADJ_A, bool ADJ_B> 243 struct SparseTensorDenseMatMulFunctor<CPUDevice, T, Tindices, ADJ_A, ADJ_B> { 248 typename TTypes<Tindices>::ConstMatrix a_indices, in Compute() [all …]
|
D | gather_nd_op_cpu_impl.h | 45 const Index slice_size, typename TTypes<Index>::ConstMatrix Tindices, in GatherNdSliceGenerator() argument 49 Tindices_(Tindices), in GatherNdSliceGenerator() 101 typename TTypes<Index>::ConstMatrix Tindices, 105 const Eigen::DenseIndex batch_size = Tindices.dimension(0); 115 slice_size, Tindices, Tparams, Tout, &error_loc); 132 typename TTypes<Index>::ConstMatrix Tindices, \
|
D | scatter_nd_op_cpu_impl.h | 96 typename TTypes<Index, 2>::ConstTensor Tindices, 103 const Eigen::DenseIndex batch_size = Tindices.dimension(0); 119 const Index ix_d = internal::SubtleMustCopy(Tindices(loc, dim)); 147 typename TTypes<Index, 2>::ConstTensor Tindices, \ 178 typename TTypes<Index, 2>::ConstTensor Tindices, 185 const Eigen::DenseIndex batch_size = Tindices.dimension(0); 201 const Index ix_d = internal::SubtleMustCopy(Tindices(loc, dim)); 229 typename TTypes<Index, 2>::ConstTensor Tindices, \
|
D | gather_nd_op_gpu.cu.cc | 72 typename TTypes<Index>::ConstMatrix Tindices, in operator ()() 74 const int64 indices_size = Tindices.dimension(1); in operator ()() 92 Tparams.data(), Tindices.data(), Tout.data(), batch_strides, in operator ()()
|
D | training_ops.cc | 809 #define REGISTER_KERNELS(T, Tindices) \ argument 813 .TypeConstraint<Tindices>("Tindices"), \ 814 SparseApplyAdadeltaOp<T, Tindices>); \ 818 .TypeConstraint<Tindices>("Tindices"), \ 819 SparseApplyAdadeltaOp<T, Tindices>); 1028 #define REGISTER_KERNELS(T, Tindices) \ argument 1032 .TypeConstraint<Tindices>("Tindices"), \ 1033 SparseApplyProximalGradientDescentOp<T, Tindices>); \ 1037 .TypeConstraint<Tindices>("Tindices"), \ 1038 SparseApplyProximalGradientDescentOp<T, Tindices>); [all …]
|
D | sparse_tensor_dense_matmul_op.h | 28 template <typename Device, typename T, typename Tindices, bool ADJ_A, 33 typename TTypes<Tindices>::ConstMatrix a_indices,
|
D | scatter_nd_op_gpu.cu.cc | 117 typename TTypes<Index, 2>::ConstTensor Tindices, in operator ()() 125 const Eigen::DenseIndex batch_size = Tindices.dimension(0); in operator ()() 142 Tindices.data(), Tupdates.data(), Toutput.data(), output_shape_prefix, in operator ()()
|
D | listdiff_op.cc | 76 auto Tindices = indices->vec<Tidx>(); in Compute() local 87 Tindices(p) = i; in Compute()
|
D | gather_functor.cc | 32 typename TTypes<Index>::ConstFlat Tindices, \
|
D | gather_nd_op.h | 40 typename TTypes<Index>::ConstMatrix Tindices,
|
D | scatter_nd_op.h | 55 typename TTypes<Index, 2>::ConstTensor Tindices,
|
D | gather_nd_op.cc | 213 typename TTypes<Index>::ConstMatrix Tindices, \
|
D | scatter_nd_op.cc | 520 typename TTypes<Index, 2>::ConstTensor Tindices, \
|
/external/tensorflow/tensorflow/core/ops/compat/ |
D | ops_history.v1.pbtxt | 20392 type_attr: "Tindices" 20410 name: "Tindices" 20428 type_attr: "Tindices" 20439 name: "Tindices" 20457 type_attr: "Tindices" 20472 name: "Tindices" 42531 type_attr: "Tindices" 42549 name: "Tindices" 42568 type_attr: "Tindices" 42597 name: "Tindices" [all …]
|
D | ops_history.v0.pbtxt | 13127 type_attr: "Tindices" 13138 name: "Tindices" 13156 type_attr: "Tindices" 13174 name: "Tindices" 13192 type_attr: "Tindices" 13203 name: "Tindices" 26487 type_attr: "Tindices" 26512 name: "Tindices" 26561 type_attr: "Tindices" 26586 name: "Tindices" [all …]
|
/external/tensorflow/tensorflow/core/ops/ |
D | ops.pbtxt | 9521 type_attr: "Tindices" 9539 name: "Tindices" 9557 type_attr: "Tindices" 9568 name: "Tindices" 9586 type_attr: "Tindices" 9601 name: "Tindices" 20851 type_attr: "Tindices" 20869 name: "Tindices" 20888 type_attr: "Tindices" 20920 name: "Tindices" [all …]
|
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
D | gpu_supported_ops.md | 86 `Gather` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double… 87 `GatherV2` | `Taxis={int32,int64}`<br>`Tindices={int32,int64}`<br>`Tpara… 176 `ResourceGather` | `Tindices={int32,int64}`<br>`dtype={complex64,double,float,… 246 `UnsortedSegmentSum` | `Tnumsegments={int32,int64}`<br>`Tindices={int32,int64}`<br…
|
D | cpu_supported_ops.md | 86 `Gather` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double… 87 `GatherV2` | `Taxis={int32,int64}`<br>`Tindices={int32,int64}`<br>`Tpara… 179 `ResourceGather` | `Tindices={int32,int64}`<br>`dtype={complex64,double,float,… 250 `UnsortedSegmentSum` | `Tnumsegments={int32,int64}`<br>`Tindices={int32,int64}`<br…
|