Searched refs:MlirHloBuilder (Results 1 – 5 of 5) sorted by relevance
70 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 …]
45 class MlirHloBuilder : public XlaBuilder {49 explicit MlirHloBuilder(mlir::FuncOp func) in MlirHloBuilder() function57 MlirHloBuilder(std::string name, mlir::OpBuilder builder, mlir::Location loc) in MlirHloBuilder() function60 MlirHloBuilder(const MlirHloBuilder&) = delete;61 MlirHloBuilder& operator=(const MlirHloBuilder&) = delete;63 ~MlirHloBuilder() override;
68 MlirHloBuilder xla_builder_;
341 ::xla::MlirHloBuilder hlo_builder_;
117 friend class MlirHloBuilder; variable