Home
last modified time | relevance | path

Searched refs:Tindices (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_tensor_dense_matmul_op_gpu.cu.cc30 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 …]
Dsparse_tensor_dense_matmul_op.cc32 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 …]
Dgather_nd_op_cpu_impl.h45 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, \
Dscatter_nd_op_cpu_impl.h96 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, \
Dgather_nd_op_gpu.cu.cc72 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 ()()
Dtraining_ops.cc809 #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 …]
Dsparse_tensor_dense_matmul_op.h28 template <typename Device, typename T, typename Tindices, bool ADJ_A,
33 typename TTypes<Tindices>::ConstMatrix a_indices,
Dscatter_nd_op_gpu.cu.cc117 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 ()()
Dlistdiff_op.cc76 auto Tindices = indices->vec<Tidx>(); in Compute() local
87 Tindices(p) = i; in Compute()
Dgather_functor.cc32 typename TTypes<Index>::ConstFlat Tindices, \
Dgather_nd_op.h40 typename TTypes<Index>::ConstMatrix Tindices,
Dscatter_nd_op.h55 typename TTypes<Index, 2>::ConstTensor Tindices,
Dgather_nd_op.cc213 typename TTypes<Index>::ConstMatrix Tindices, \
Dscatter_nd_op.cc520 typename TTypes<Index, 2>::ConstTensor Tindices, \
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt20392 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 …]
Dops_history.v0.pbtxt13127 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/
Dops.pbtxt9521 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/
Dgpu_supported_ops.md86 `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…
Dcpu_supported_ops.md86 `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…