Searched defs:output_batch_shape (Results 1 – 3 of 3) sorted by relevance
72 const TensorShape& output_batch_shape() const { return output_batch_shape_; } in output_batch_shape() function
74 ShapeHandle output_batch_shape; in BandedTriangularSolveShapeFn() local140 ShapeHandle output_batch_shape; in MatrixTriangularSolveShapeFn() local
331 ShapeHandle output_batch_shape; in BatchMatMulV2Shape() local