Home
last modified time | relevance | path

Searched defs:indices_size (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops_gpu.cu.cc39 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
H A Dgather_functor_gpu.cu.h36 int64 gather_dim_size, int64 indices_size, in GatherOpKernel()
110 int64 indices_size, int64 slice_size, in LaunchGatherKernel()
H A Dgather_functor.h44 const SliceIndex indices_size = static_cast<SliceIndex>(indices.dimension(0)); in HandleCopies() local
119 const int64_t indices_size = indices.size(); in operator() local
H A Dgather_functor_batched.h46 const SliceIndex indices_size = in HandleCopiesBatched() local
133 const int64_t indices_size = indices.size(); // Includes the batch_size. in operator() local
H A Dgather_functor_batched_gpu.cu.h37 int64 gather_dim_size, int64 indices_size, in GatherOpKernel()
125 int64 gather_dim_size, int64 indices_size, in LaunchGatherKernel()
H A Dgather_nd_op_gpu.cu.cc33 const Eigen::array<int64, IXDIM> batch_indices, const int64 indices_size, in GatherSliceOpKernel()
74 const int64 indices_size = Tindices.dimension(1); in operator ()() local
H A Ddynamic_partition_op_gpu.cu.cc451 int32 indices_size = outs[p]->dim_size(0); in GatherSlices() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtfrt_ops.cc56 int32_t indices_size = in verify() local
/aosp_15_r20/external/perfetto/src/trace_processor/db/column/
H A Dutils.cc113 uint32_t indices_size) { in CanReturnEarly()
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad.cc662 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/perception/
H A Dmax_pool_with_argmax.cc196 TfLiteIntArray* indices_size = TfLiteIntArrayCopy(output_size); in Prepare() local
/aosp_15_r20/external/mesa3d/src/mesa/main/
H A Dglthread_draw.c1068 int indices_size = sizeof(indices[0]) * real_draw_count; in multi_draw_elements_async() local