Home
last modified time | relevance | path

Searched defs:x_batch_indices (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmatmul_bcast.h80 const std::vector<int64_t>& x_batch_indices() const { in x_batch_indices() function
H A Dbcast.h389 const std::vector<int64_t>& x_batch_indices() const { in x_batch_indices() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_op_impl.h104 const auto& x_batch_indices = bcast.x_batch_indices(); in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_triangular_solve_op_impl.h83 const auto& x_batch_indices = bcast.x_batch_indices(); in Run() local
H A Dbanded_triangular_solve_op.cc66 const auto& x_batch_indices = bcast.x_batch_indices(); in Run() local