Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dlinalg_ops.cc74 ShapeHandle output_batch_shape; in BandedTriangularSolveShapeFn() local
140 ShapeHandle output_batch_shape; in MatrixTriangularSolveShapeFn() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmatmul_bcast.h72 const TensorShape& output_batch_shape() const { return output_batch_shape_; } in output_batch_shape() function
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc331 ShapeHandle output_batch_shape; in BatchMatMulV2Shape() local