HomeSort by relevance Sort by last modified time
    Searched full:tindices (Results 1 - 25 of 37) sorted by null

1 2

  /external/tensorflow/tensorflow/core/ops/
state_ops.cc 135 .Input("indices: Tindices")
139 .Attr("Tindices: {int32, int64}")
145 .Input("indices: Tindices")
149 .Attr("Tindices: {int32, int64}")
155 .Input("indices: Tindices")
159 .Attr("Tindices: {int32, int64}")
165 .Input("indices: Tindices")
169 .Attr("Tindices: {int32, int64}")
175 .Input("indices: Tindices")
179 .Attr("Tindices: {int32, int64}"
    [all...]
training_ops.cc 121 .Input("indices: Tindices")
124 .Attr("Tindices: {int32, int64}")
148 .Input("indices: Tindices")
150 .Attr("Tindices: {int32, int64}")
196 .Input("indices: Tindices")
199 .Attr("Tindices: {int32, int64}")
227 .Input("indices: Tindices")
229 .Attr("Tindices: {int32, int64}")
318 .Input("indices: Tindices")
321 .Attr("Tindices: {int32, int64}"
    [all...]
resource_variable_ops.cc 147 .Input("indices: Tindices")
151 .Attr("Tindices: {int32,int64}")
172 .Input("indices: Tindices")
175 .Attr("Tindices: {int32, int64}")
194 .Input("indices: Tindices")
197 .Attr("Tindices: {int32, int64}")
sparse_ops.cc 86 .Input("a_indices: Tindices")
92 .Attr("Tindices: {int32,int64} = DT_INT64")
193 .Input("sparse_indices: Tindices")
194 .Input("output_shape: Tindices")
200 .Attr("Tindices: {int32, int64}")
370 .Input("a_indices: Tindices")
372 .Input("a_shape: Tindices")
376 .Attr("Tindices: {int32, int64}")
math_ops.cc 1010 .Input("segment_ids: Tindices")
1013 .Attr("Tindices: {int32,int64}")
1018 .Input("segment_ids: Tindices")
1021 .Attr("Tindices: {int32,int64}")
1026 .Input("segment_ids: Tindices")
1029 .Attr("Tindices: {int32,int64}")
1034 .Input("segment_ids: Tindices")
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
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,
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,
84 SparseTensorDenseMatMulKernel<T, Tindices, ADJ_A, ADJ_B>
95 #define DEFINE(T, Tindices) \
97 GPUDevice, T, Tindices, false, false>; \
99 GPUDevice, T, Tindices, false, true>; \
101 GPUDevice, T, Tindices, true, false>;
    [all...]
gather_nd_op.h 34 // Performs a slice gather op on (Tparams, Tindices), writing to Tout.
35 // Returns an index to Tindices if the value at that index is out of range.
36 // Returns -1 if all values of Tindices are in range.
40 typename TTypes<Index>::ConstMatrix Tindices,
sparse_tensor_dense_matmul_op.cc 32 template <typename Device, typename T, typename Tindices>
138 Device, T, Tindices, ADJ_A, \
140 a_indices->matrix<Tindices>(), a_values->vec<T>(), \
163 .TypeConstraint<TypeIndex>("Tindices") \
180 #define DECLARE_GPU_SPEC(T, Tindices, ADJ_A, ADJ_B) \
183 GPUDevice, T, Tindices, ADJ_A, \
185 TTypes<Tindices>::ConstMatrix a_indices, \
189 GPUDevice, T, Tindices, ADJ_A, ADJ_B>;
213 .TypeConstraint<TypeIndex>("Tindices") \
242 template <typename T, typename Tindices, bool ADJ_A, bool ADJ_B
    [all...]
scatter_nd_op_cpu_impl.h 96 typename TTypes<Index, 2>::ConstTensor Tindices,
100 // otherwise it is the location of an OOB index in 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,
182 // otherwise it is the location of an OOB index in 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, \
    [all...]
gather_nd_op_cpu_impl.h 45 const Index slice_size, typename TTypes<Index>::ConstMatrix Tindices,
49 Tindices_(Tindices),
101 typename TTypes<Index>::ConstMatrix Tindices,
105 const Eigen::DenseIndex batch_size = Tindices.dimension(0);
115 slice_size, Tindices, Tparams, Tout, &error_loc);
122 // otherwise it returns the location of an OOB index in Tindices.
132 typename TTypes<Index>::ConstMatrix Tindices, \
sparse_tensor_dense_matmul_op.h 28 template <typename Device, typename T, typename Tindices, bool ADJ_A,
33 typename TTypes<Tindices>::ConstMatrix a_indices,
training_ops.cc     [all...]
listdiff_op.cc 76 auto Tindices = indices->vec<Tidx>();
87 Tindices(p) = i;
gather_functor.cc 32 typename TTypes<Index>::ConstFlat Tindices, \
gather_nd_op_gpu.cu.cc 72 typename TTypes<Index>::ConstMatrix Tindices,
74 const int64 indices_size = Tindices.dimension(1);
92 Tparams.data(), Tindices.data(), Tout.data(), batch_strides,
scatter_nd_op_gpu.cu.cc 117 typename TTypes<Index, 2>::ConstTensor Tindices,
125 const Eigen::DenseIndex batch_size = Tindices.dimension(0);
142 Tindices.data(), Tupdates.data(), Toutput.data(), output_shape_prefix,
gather_op.cc 126 .TypeConstraint<index_type>("Tindices"), \
131 .TypeConstraint<index_type>("Tindices") \
scatter_nd_op.cc 178 .TypeConstraint<index_type>("Tindices") \
188 .TypeConstraint<index_type>("Tindices"), \
197 .TypeConstraint<index_type>("Tindices") \
520 typename TTypes<Index, 2>::ConstTensor Tindices, \
scatter_nd_op.h 55 typename TTypes<Index, 2>::ConstTensor Tindices,
  /external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/
reduce_slice_ops.cc 68 .Input("indices: Tindices")
72 .Attr("Tindices: {int32,int64}")
107 Tindices: the type of indices, must be int32 or int64.
113 .Input("indices: Tindices")
117 .Attr("Tindices: {int32,int64}")
164 Tindices: the type of indices, must be int32 or int64.
170 .Input("indices: Tindices")
174 .Attr("Tindices: {int32,int64}")
221 Tindices: the type of indices, must be int32 or int64.
227 .Input("indices: Tindices")
    [all...]
  /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/
reduce_slice_ops.cc 156 .TypeConstraint<index_type>("Tindices"), \
162 .TypeConstraint<index_type>("Tindices"), \
170 .TypeConstraint<index_type>("Tindices"), \
176 .TypeConstraint<index_type>("Tindices"), \
203 .TypeConstraint<index_type>("Tindices"), \
210 .TypeConstraint<index_type>("Tindices"), \
217 .TypeConstraint<index_type>("Tindices"), \
224 .TypeConstraint<index_type>("Tindices"), \
  /external/tensorflow/tensorflow/core/ops/compat/
ops_history.v0.pbtxt     [all...]
  /external/eigen/demos/opengl/
icosphere.cpp 29 static GLint tindices[20][3] = { variable
46 indices.push_back(tindices[i][k]);
  /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
cpu_supported_ops.md 86 `Gather` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double,float,int32,int64,uint32,uint64}`
87 `GatherV2` | `Taxis={int32,int64}`<br>`Tindices={int32,int64}`<br>`Tparams={bool,complex64,double,float,int32,int64,uint32,uint64}`
179 `ResourceGather` | `Tindices={int32,int64}`<br>`dtype={complex64,double,float,int32,int64,uint32,uint64}`
250 `UnsortedSegmentSum` | `Tnumsegments={int32,int64}`<br>`Tindices={int32,int64}`<br>`T={complex64,double,float,int32,int64,uint32,uint64}`
gpu_supported_ops.md 86 `Gather` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double,float,int32,int64,uint32,uint64}`
87 `GatherV2` | `Taxis={int32,int64}`<br>`Tindices={int32,int64}`<br>`Tparams={bool,complex64,double,float,int32,int64,uint32,uint64}`
176 `ResourceGather` | `Tindices={int32,int64}`<br>`dtype={complex64,double,float,int32,int64,uint32,uint64}`
246 `UnsortedSegmentSum` | `Tnumsegments={int32,int64}`<br>`Tindices={int32,int64}`<br>`T={complex64,double,float,int32,int64,uint32,uint64}`

Completed in 575 milliseconds

1 2