Searched refs:hlo_module_proto (Results 1 – 4 of 4) sorted by relevance
25 xla::HloModuleProto* hlo_module_proto, in ConvertHloToMlirHlo() argument29 .Import(*hlo_module_proto); in ConvertHloToMlirHlo()
1664 xla::HloModuleProto* hlo_module_proto, in AddDynamicParameterBindings() argument1670 hlo_module_proto->mutable_dynamic_parameter_binding(); in AddDynamicParameterBindings()
75 string hlo_module_proto; member675 "'--hlo_snapshot', '--hlo_text', '--hlo_module_proto' flags.";750 tensorflow::Flag("hlo_module_proto", &opts.hlo_module_proto,
1784 const auto& hlo_module_proto = result.computation->proto(); in TEST_F() local1785 ASSERT_EQ(hlo_module_proto.computations_size(), 1); in TEST_F()1786 const auto& hlo_computation_proto = hlo_module_proto.computations(0); in TEST_F()