Home
last modified time | relevance | path

Searched defs:a_batch_dims (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dsparse_csr_matrix_ops.cc277 ShapeHandle a_batch_dims; in __anona88575120702() local
430 ShapeHandle a_batch_dims; in __anona88575120a02() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dqr.cc93 absl::Span<const int64_t> a_batch_dims = in ProductOfElementaryHouseholderReflectors() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc1995 absl::Span<const int64_t> a_batch_dims(a.dimensions()); in InferTriangularSolveShape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc361 ShapeHandle a_batch_dims; in BatchMatMulShape() local