Home
last modified time | relevance | path

Searched defs:b_batch_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc278 ShapeHandle b_batch_dims; in __anonc9a48f920702() local
431 ShapeHandle b_batch_dims; in __anonc9a48f920a02() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc1996 absl::Span<const int64_t> b_batch_dims(b.dimensions()); in InferTriangularSolveShape() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc362 ShapeHandle b_batch_dims; in BatchMatMulShape() local