/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}`
|