Searched refs:ReshapeInternal (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 200 StatusOr<XlaOp> ReshapeInternal(const Shape& shape, XlaOp operand,
|
| D | mlir_hlo_builder.cc | 363 StatusOr<XlaOp> MlirHloBuilder::ReshapeInternal(const Shape& shape, in ReshapeInternal() function in xla::MlirHloBuilder
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 610 ReshapeInternal(reshaped_shape, operand, /*inferred_dimension=*/-1)); in AddBroadcastSequence()
940 StatusOr<XlaOp> XlaBuilder::ReshapeInternal(const Shape& shape, XlaOp operand, in ReshapeInternal() function in xla::XlaBuilder
1126 return ReshapeInternal(shape, transposed, inferred_dimension); in Reshape()
1143 return ReshapeInternal(shape, operand, inferred_dimension); in Reshape()
|
| D | xla_builder.h | 967 virtual StatusOr<XlaOp> ReshapeInternal(const Shape& shape, XlaOp operand,
|