Home
last modified time | relevance | path

Searched refs:CreateFromProto (Results 1 – 25 of 72) 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/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.cc68 /* static */ StatusOr<HloModuleGroup> HloModuleGroup::CreateFromProto( in CreateFromProto() function in xla::HloModuleGroup
81 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_proto_util.cc47 HloModule::CreateFromProto(proto, module_config)); in CreateModuleFromProto()
Ddynamic_parameter_binding.h105 static StatusOr<DynamicParameterBinding> CreateFromProto(
Dhlo_module_group.h84 static StatusOr<HloModuleGroup> CreateFromProto(
Dhlo_input_output_alias_config.h91 static StatusOr<HloInputOutputAliasConfig> CreateFromProto(
Dhlo_module.cc389 StatusOr<std::unique_ptr<HloModule>> HloModule::CreateFromProto( in CreateFromProto() function in xla::HloModule
429 HloComputation::CreateFromProto(computation_proto, computation_map, in CreateFromProto()
464 HloInputOutputAliasConfig::CreateFromProto( in CreateFromProto()
470 DynamicParameterBinding::CreateFromProto( in CreateFromProto()
479 HloSchedule::CreateFromProto(module.get(), proto.schedule())); in CreateFromProto()
Dhlo_module_test.cc346 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
367 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
433 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
Drng_bit_generator_expander.cc92 TF_ASSIGN_OR_RETURN(auto new_module, HloModule::CreateFromProto( in GetGeneratorComputation()
Ddynamic_parameter_binding_test.cc41 DynamicParameterBinding::CreateFromProto(proto)); in SerializeAndDeserialize()
Dcompile_only_service.cc128 HloModule::CreateFromProto(instance.computation, *module_config)); in CompileAheadOfTime()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout.cc53 /* static */ Layout Layout::CreateFromProto(const LayoutProto& proto) { in CreateFromProto() function in xla::Layout
60 *layout.add_tiles() = Tile::CreateFromProto(tile_proto); in CreateFromProto()
Dlayout.h40 static Tile CreateFromProto(const TileProto& tile_proto) { in CreateFromProto() function
104 static Layout CreateFromProto(const LayoutProto& proto);
Dliteral_test.cc1549 TF_ASSERT_OK_AND_ASSIGN(Literal literal, Literal::CreateFromProto(p)); in TEST_F()
1595 TF_ASSERT_OK_AND_ASSIGN(Literal literal, Literal::CreateFromProto(p)); in TEST_F()
1617 TF_ASSERT_OK_AND_ASSIGN(Literal literal, Literal::CreateFromProto(p)); in TEST_F()
1901 return Literal::CreateFromProto(literal.ToProto()).ValueOrDie(); in TEST_F()
1923 Status status = Literal::CreateFromProto(proto).status(); in TEST_F()
1934 Literal::CreateFromProto(proto, /*prohibit_empty_literal=*/false) in TEST_F()
1948 Literal::CreateFromProto(proto, /*prohibit_empty_literal=*/false) in TEST_F()
1959 Status status = Literal::CreateFromProto(proto).status(); in TEST_F()
1971 Status status = Literal::CreateFromProto(proto).status(); in TEST_F()
1984 Status status = Literal::CreateFromProto(proto).status(); in TEST_F()
[all …]
Dlayout_test.cc105 EXPECT_EQ(layout, Layout::CreateFromProto(layout.ToProto())); in TEST_F()
/external/perfetto/src/tracing/
Dtraced_value.cc31 return TracedValue::CreateFromProto(annotation, event_context); in CreateTracedValueFromProto()
37 TracedValue TracedValue::CreateFromProto( in CreateFromProto() function in perfetto::TracedValue
/external/tensorflow/tensorflow/compiler/xla/tools/
Dshow_literal.cc43 xla::Literal literal = xla::Literal::CreateFromProto(literal_proto).value(); in main()
Ddumped_computation_to_text.cc81 HloModule::CreateFromProto(computation.proto(), config).value(); in RealMain()
Dhlo_module_loader.cc114 module, HloModule::CreateFromProto(proto.hlo().hlo_module(), config)); in LoadModuleFromData()
/external/tensorflow/tensorflow/core/profiler/utils/
Dhlo_proto_to_module.cc39 xla::HloModule::CreateFromProto(module_proto, config)); in ConvertHloProtoToModule()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_module_importer.cc67 TF_ASSIGN_OR_RETURN(auto module, xla::HloModule::CreateFromProto( in Import()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc64 return Literal::CreateFromProto(*response.mutable_literal()); in Transfer()
146 return Literal::CreateFromProto(response.literal()); in TransferFromOutfeed()
208 return Literal::CreateFromProto(response.literal()); in ComputeConstant()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc172 xla::Layout::CreateFromProto(bitcast_config.result_layout()); in GetSourceIndexOfBitcast()
174 xla::Layout::CreateFromProto(bitcast_config.source_layout()); in GetSourceIndexOfBitcast()

123