Home
last modified time | relevance | path

Searched refs:MlirHloBuilder (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc70 MlirHloBuilder::~MlirHloBuilder() = default;
72 StatusOr<XlaOp> MlirHloBuilder::MakeXlaOp(mlir::Value val) { in MakeXlaOp()
84 XlaOp MlirHloBuilder::ConstantLiteral(const LiteralSlice& literal) { in ConstantLiteral()
93 StatusOr<XlaOp> MlirHloBuilder::ConvGeneralDilatedInternal( in ConvGeneralDilatedInternal()
119 StatusOr<XlaOp> MlirHloBuilder::FftInternal( in FftInternal()
131 StatusOr<XlaOp> MlirHloBuilder::CustomCallInternal( in CustomCallInternal()
155 StatusOr<XlaOp> MlirHloBuilder::ReduceInternal( in ReduceInternal()
172 StatusOr<XlaOp> MlirHloBuilder::ReduceWindowInternal( in ReduceWindowInternal()
201 XlaOp MlirHloBuilder::Iota(const Shape& shape, int64 iota_dimension) { in Iota()
213 StatusOr<XlaOp> MlirHloBuilder::BitcastConvertTypeInternal(const Shape& shape, in BitcastConvertTypeInternal()
[all …]
Dmlir_hlo_builder.h45 class MlirHloBuilder : public XlaBuilder {
49 explicit MlirHloBuilder(mlir::FuncOp func) in MlirHloBuilder() function
57 MlirHloBuilder(std::string name, mlir::OpBuilder builder, mlir::Location loc) in MlirHloBuilder() function
60 MlirHloBuilder(const MlirHloBuilder&) = delete;
61 MlirHloBuilder& operator=(const MlirHloBuilder&) = delete;
63 ~MlirHloBuilder() override;
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dmlir_hlo_builder_test.cc68 MlirHloBuilder xla_builder_;
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc341 ::xla::MlirHloBuilder hlo_builder_;
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h117 friend class MlirHloBuilder; variable