Searched defs:output_batch_shape (Results 1 – 3 of 3) sorted by relevance
42 const TensorShape& output_batch_shape() const { return output_shape_; } in output_batch_shape() function
97 ShapeHandle output_batch_shape; in MatrixTriangularSolveShapeFn() local
440 ShapeHandle output_batch_shape; in BatchMatMulV2Shape() local