Searched defs:indices_size (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_functor_gpu.cu.h | 36 int64 gather_dim_size, int64 indices_size, in GatherOpKernel()
110 int64 indices_size, int64 slice_size, in LaunchGatherKernel()
|
| D | gather_functor.h | 44 const SliceIndex indices_size = static_cast<SliceIndex>(indices.dimension(0)); in HandleCopies() local
119 const int64_t indices_size = indices.size(); in operator() local
|
| D | gather_functor_batched.h | 46 const SliceIndex indices_size = in HandleCopiesBatched() local
133 const int64_t indices_size = indices.size(); // Includes the batch_size. in operator() local
|
| D | gather_nd_op_gpu.cu.cc | 33 const Eigen::array<int64, IXDIM> batch_indices, const int64 indices_size, in GatherSliceOpKernel()
74 const int64 indices_size = Tindices.dimension(1); in operator ()() local
|
| D | gather_functor_batched_gpu.cu.h | 37 int64 gather_dim_size, int64 indices_size, in GatherOpKernel()
125 int64 gather_dim_size, int64 indices_size, in LaunchGatherKernel()
|
| D | training_ops_gpu.cu.cc | 39 Tindex indices_size, bool update_slots) { in SparseApplyAdagradKernel()
78 Tindex indices_size) { in SparseApplyProximalAdagradKernel()
120 Tindex indices_size, in SparseApplyFtrlKernel()
225 Tindex updates_size, Tindex indices_size) { in SparseApplyKerasMomentumKernel()
441 const Tindex indices_size = indices.size(); in operator ()() local
502 const Tindex indices_size = indices.size(); in operator ()() local
712 const Tindex indices_size = indices.size(); in operator ()() local
788 const Tindex indices_size = indices.size(); in operator ()() local
805 Tindex updates_size, Tindex indices_size) { in SparseApplyAdadeltaKernel()
854 const Tindex indices_size = indices.size(); in operator ()() local
|
| D | dynamic_partition_op_gpu.cu.cc | 451 int32 indices_size = outs[p]->dim_size(0); in GatherSlices() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tfrt_ops.cc | 56 int32_t indices_size = in verify() local
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | max_pool_with_argmax.cc | 196 TfLiteIntArray* indices_size = TfLiteIntArrayCopy(output_size); in Prepare() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 662 auto indices_size = ExpandDims(scope, Size(scope, indices), 0); in BatchGatherGrad() local
711 auto indices_size = ExpandDims(scope, Size(scope, indices), 0); in GatherV2Grad() local
|
| /external/mesa3d/src/mesa/main/ |
| D | glthread_draw.c | 779 int indices_size = sizeof(indices[0]) * draw_count; in multi_draw_elements_async() local
|