Home
last modified time | relevance | path

Searched refs:CreateFromProto (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_topk_test.cc41 auto module, HloModule::CreateFromProto(xla_computation.proto(), config)); in TEST_F()
68 auto module, HloModule::CreateFromProto(xla_computation.proto(), config)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout.cc48 /* static */ Layout Layout::CreateFromProto(const LayoutProto& proto) { in CreateFromProto() function in xla::Layout
56 *layout.add_tiles() = Tile::CreateFromProto(tile_proto); in CreateFromProto()
Dlayout.h40 static Tile CreateFromProto(const TileProto& tile_proto) { in CreateFromProto() function
102 static Layout CreateFromProto(const LayoutProto& proto);
Dliteral_test.cc1541 TF_ASSERT_OK_AND_ASSIGN(Literal literal, Literal::CreateFromProto(p)); in TEST_F()
1587 TF_ASSERT_OK_AND_ASSIGN(Literal literal, Literal::CreateFromProto(p)); in TEST_F()
1609 TF_ASSERT_OK_AND_ASSIGN(Literal literal, Literal::CreateFromProto(p)); in TEST_F()
1892 return Literal::CreateFromProto(literal.ToProto()).ValueOrDie(); in TEST_F()
1914 Status status = Literal::CreateFromProto(proto).status(); in TEST_F()
1925 Literal::CreateFromProto(proto, /*prohibit_empty_literal=*/false) in TEST_F()
1939 Literal::CreateFromProto(proto, /*prohibit_empty_literal=*/false) in TEST_F()
1950 Status status = Literal::CreateFromProto(proto).status(); in TEST_F()
1962 Status status = Literal::CreateFromProto(proto).status(); in TEST_F()
1975 Status status = Literal::CreateFromProto(proto).status(); in TEST_F()
[all …]
Dlayout_test.cc102 EXPECT_EQ(layout, Layout::CreateFromProto(layout.ToProto())); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner_interface.cc39 HloModule::CreateFromProto(proto.hlo_module(), config)); in HloProtoToModule()
85 return HloModule::CreateFromProto(module_proto, module_config); in ReadModuleFromModuleBinaryProtofile()
Dhlo_module_group.cc78 /* static */ StatusOr<HloModuleGroup> HloModuleGroup::CreateFromProto( in CreateFromProto() function in xla::HloModuleGroup
91 HloModule::CreateFromProto(module_proto, module_configs[i])); in CreateFromProto()
Dhlo_module_group_test.cc58 HloModuleGroup::CreateFromProto( in TEST_F()
103 HloModuleGroup::CreateFromProto( in TEST_F()
Dhlo_module.cc330 StatusOr<std::unique_ptr<HloModule>> HloModule::CreateFromProto( in CreateFromProto() function in xla::HloModule
370 HloComputation::CreateFromProto(computation_proto, computation_map, in CreateFromProto()
406 HloInputOutputAliasConfig::CreateFromProto( in CreateFromProto()
412 DynamicParameterBinding::CreateFromProto( in CreateFromProto()
421 HloSchedule::CreateFromProto(module.get(), proto.schedule())); in CreateFromProto()
Dhlo_module_test.cc220 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
241 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
307 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
Dhlo_module_group.h78 static StatusOr<HloModuleGroup> CreateFromProto(
Dhlo_proto_util.cc45 HloModule::CreateFromProto(proto, module_config)); in CreateModuleFromProto()
Ddynamic_parameter_binding.h103 static StatusOr<DynamicParameterBinding> CreateFromProto(
Dhlo_input_output_alias_config.h91 static StatusOr<HloInputOutputAliasConfig> CreateFromProto(
Ddynamic_parameter_binding_test.cc41 DynamicParameterBinding::CreateFromProto(proto)); in SerializeAndDeserialize()
Drng_bit_generator_expander.cc92 TF_ASSIGN_OR_RETURN(auto new_module, HloModule::CreateFromProto( in GetGeneratorComputation()
Dcompile_only_service.cc111 HloModule::CreateFromProto(instance.computation, *module_config)); in CompileAheadOfTime()
Dhlo_schedule.h112 static StatusOr<HloSchedule> CreateFromProto(const HloModule* module,
Ddynamic_parameter_binding.cc64 StatusOr<DynamicParameterBinding> DynamicParameterBinding::CreateFromProto( in CreateFromProto() function in xla::DynamicParameterBinding
/external/perfetto/src/tracing/
Dtraced_value.cc29 return TracedValue::CreateFromProto(context); in CreateTracedValueFromProto()
35 TracedValue TracedValue::CreateFromProto( in CreateFromProto() function in perfetto::TracedValue
/external/tensorflow/tensorflow/compiler/xla/tools/
Dshow_literal.cc44 xla::Literal::CreateFromProto(literal_proto).ConsumeValueOrDie(); in main()
Ddumped_computation_to_text.cc79 HloModule::CreateFromProto(computation.proto(), config) in RealMain()
Dhlo_module_loader.cc111 module, HloModule::CreateFromProto(proto.hlo().hlo_module(), config)); in LoadModuleFromData()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_module_importer.cc61 TF_ASSIGN_OR_RETURN(auto module, xla::HloModule::CreateFromProto( in Import()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc65 return Literal::CreateFromProto(*response.mutable_literal()); in Transfer()
147 return Literal::CreateFromProto(response.literal()); in TransferFromOutfeed()
209 return Literal::CreateFromProto(response.literal()); in ComputeConstant()

123