Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast.h42 const TensorShape& output_batch_shape() const { return output_shape_; } in output_batch_shape() function
/external/tensorflow/tensorflow/core/ops/
Dlinalg_ops.cc97 ShapeHandle output_batch_shape; in MatrixTriangularSolveShapeFn() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc440 ShapeHandle output_batch_shape; in BatchMatMulV2Shape() local