Searched refs:ReshapeInternal (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 211 StatusOr<XlaOp> ReshapeInternal(const Shape& shape, XlaOp operand,
|
| D | mlir_hlo_builder.cc | 484 StatusOr<XlaOp> MlirHloBuilder::ReshapeInternal(const Shape& shape, in ReshapeInternal() function in xla::MlirHloBuilder
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 673 ReshapeInternal(reshaped_shape, operand, /*inferred_dimension=*/-1)); in AddBroadcastSequence()
1006 StatusOr<XlaOp> XlaBuilder::ReshapeInternal(const Shape& shape, XlaOp operand, in ReshapeInternal() function in xla::XlaBuilder
1193 return ReshapeInternal(shape, transposed, inferred_dimension); in Reshape()
1210 return ReshapeInternal(shape, operand, inferred_dimension); in Reshape()
|
| D | xla_builder.h | 1001 virtual StatusOr<XlaOp> ReshapeInternal(const Shape& shape, XlaOp operand,
|