Searched defs:hlo_proto (Results 1 – 14 of 14) sorted by relevance
32 HloProto hlo_proto; in TEST_F() local41 HloProto hlo_proto; in TEST_F() local
55 const HloProto& hlo_proto) { in EntryComputationParameterShapes()72 const HloProto& hlo_proto) { in EntryComputationOutputShape()
31 std::unique_ptr<HloProto> hlo_proto; member
372 void set_hlo_proto(std::unique_ptr<xla::HloProto> hlo_proto) { in set_hlo_proto()376 HloProto const* hlo_proto() const { return hlo_proto_.get(); } in hlo_proto() function
295 auto hlo_proto = absl::make_unique<HloProto>(); in BuildExecutables() local782 auto hlo_proto = absl::make_unique<HloProto>(); in BuildExecutable() local
48 bool LoadHloProto(const std::string& contents, HloProto* hlo_proto) { in LoadHloProto()63 HloProto hlo_proto; in HloToMlirHloTranslateFunction() local85 HloProto hlo_proto; in HloTextToMlirHloTranslateFunction() local124 const HloProto& hlo_proto) { in HloModuleFromProto()135 ::xla::HloProto* hlo_proto, in ConvertMlirHloToHloViaBuilder()
1767 mlir::ModuleOp module, xla::HloProto* hlo_proto, bool use_tuple_args, in ConvertMlirHloToHlo()
60 HloProto hlo_proto; in RealMain() local
74 string hlo_proto; member
279 xla::HloProto hlo_proto; in CompileMlirToXlaHloViaBuilder() local
398 xla::HloProto hlo_proto; in ConvertMLIRToXlaComputation() local
91 const HloProto& hlo_proto) { in HloModuleFromProto()177 HloProto hlo_proto; in runOnOperation() local
801 xla::HloProto* hlo_proto = proto.add_hlo_protos(); in ComputeInternal() local
837 xla::HloProto hlo_proto; in Compile() local