Home
last modified time | relevance | path

Searched defs:b_indices (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_add_grad_op.cc32 const Tensor *backprop_val_grad, *a_indices, *b_indices, *sum_indices; in Compute() local
H A Dsparse_add_op.cc33 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, in Compute() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DMaxUnpoolingLayer.cpp55 const auto b_indices = static_cast<int>(indices.shape()[3]); in max_unpooling_layer_internal() local
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/
H A Dmultiplex_3_kernel.cc80 const auto b_indices = b_indices_tensor.matrix<int64_t>(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dsparse_ops.cc56 ShapeHandle b_indices; in __anone91b66370202() local
/aosp_15_r20/hardware/google/gfxstream/host/gl/glestranslator/GLcommon/
DGLEScontext.cpp1149 GLubyte* b_indices =(GLubyte *)indices; in findMaxIndex() local