Searched refs:MlirHloBuilder (Results 1 – 5 of 5) sorted by relevance
77 MlirHloBuilder::~MlirHloBuilder() = default;79 StatusOr<XlaOp> MlirHloBuilder::MakeXlaOp(mlir::Value val) { in MakeXlaOp()91 XlaOp MlirHloBuilder::ConstantLiteral(const LiteralSlice& literal) { in ConstantLiteral()100 StatusOr<XlaOp> MlirHloBuilder::ConvGeneralDilatedInternal( in ConvGeneralDilatedInternal()127 StatusOr<XlaOp> MlirHloBuilder::FftInternal( in FftInternal()142 StatusOr<XlaOp> MlirHloBuilder::CustomCallInternal( in CustomCallInternal()224 StatusOr<XlaOp> MlirHloBuilder::ReduceInternal( in ReduceInternal()242 StatusOr<XlaOp> MlirHloBuilder::ReduceWindowInternal( in ReduceWindowInternal()272 XlaOp MlirHloBuilder::Iota(const Shape& shape, int64_t iota_dimension) { in Iota()284 StatusOr<XlaOp> MlirHloBuilder::BitcastConvertTypeInternal(const Shape& shape, in BitcastConvertTypeInternal()[all …]
47 class MlirHloBuilder : public XlaBuilder {51 explicit MlirHloBuilder(mlir::func::FuncOp func) in MlirHloBuilder() function60 MlirHloBuilder(std::string name, mlir::OpBuilder builder, mlir::Location loc, in MlirHloBuilder() function67 MlirHloBuilder(const MlirHloBuilder&) = delete;68 MlirHloBuilder& operator=(const MlirHloBuilder&) = delete;70 ~MlirHloBuilder() override;
79 MlirHloBuilder xla_builder_;
481 ::xla::MlirHloBuilder hlo_builder_;
137 friend class MlirHloBuilder; variable