Home
last modified time | relevance | path

Searched refs:entry_computation_layout (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser_test.cc126 R"(HloModule module, entry_computation_layout={()->pred[2,3]{1,0}} in CreateTestCases()
138 R"(HloModule constant_s32_module, entry_computation_layout={()->s32[]} in CreateTestCases()
150 R"(HloModule ConstantF32_module, entry_computation_layout={()->f32[]} in CreateTestCases()
161 R"(HloModule ConstantF32Empty_module, entry_computation_layout={()->f32[0]{0}} in CreateTestCases()
172 R"(HloModule ConstantF32R4Empty_module, entry_computation_layout={()->f32[2,0,4,3]{3,2,1,0}} in CreateTestCases()
183 R"(HloModule Small_3x2x1x1_module, entry_computation_layout={()->f32[3,2,1,1]{3,2,1,0}} in CreateTestCases()
194 R"(HloModule IsFiniteR1F32s_module, entry_computation_layout={()->pred[6]{0}} in CreateTestCases()
206 R"(HloModule ConstantF16_module, entry_computation_layout={()->f16[]} in CreateTestCases()
217 R"(HloModule BF16, entry_computation_layout={()->bf16[]} in CreateTestCases()
228 R"(HloModule add_constants_module, entry_computation_layout={()->f32[]} in CreateTestCases()
[all …]
Dhlo_module_config.cc36 HloModuleConfig::HloModuleConfig(ComputationLayout entry_computation_layout) in HloModuleConfig() argument
37 : entry_computation_layout_(std::move(entry_computation_layout)) {} in HloModuleConfig()
Dhlo_module_util.cc167 for (int i = 0; i < module->entry_computation_layout().parameter_count(); in UpdateEntryComputationLayout()
170 module->entry_computation_layout().parameter_layout(i).shape(); in UpdateEntryComputationLayout()
175 Shape shape = module->entry_computation_layout().result_layout().shape(); in UpdateEntryComputationLayout()
Dlayout_assignment_test.cc54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts() argument
57 entry_computation_layout, in AssignLayouts()
480 ComputationLayout* entry_computation_layout) in OperandsMustBeTheSameLayoutAssignment() argument
481 : LayoutAssignment(entry_computation_layout) {} in OperandsMustBeTheSameLayoutAssignment()
1153 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1169 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1196 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1229 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1256 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1283 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
[all …]
Dhlo_module.h152 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
153 return config_.entry_computation_layout(); in entry_computation_layout()
185 h = H::combine(std::move(h), module.entry_computation_layout()); in AbslHashValue()
Dhlo_module_config.h81 explicit HloModuleConfig(ComputationLayout entry_computation_layout);
99 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
Dhlo_module.cc271 result.Append(entry_computation_layout().ToString()); in ToCord()
309 entry_computation_layout().ComputeProgramShape().ToProto(); in ToProto()
401 module_config.entry_computation_layout().parameter_count()); in CreateFromProto()
404 module_config.entry_computation_layout().parameter_layout(i).shape(); in CreateFromProto()
414 module_config.entry_computation_layout().result_layout().shape(); in CreateFromProto()
Dlocal_service.cc161 << module_config->entry_computation_layout().ToString(); in CompileExecutables()
Dservice.cc666 << module_config->entry_computation_layout().ToString(); in ExecuteGraphParallel()
879 << module_config->entry_computation_layout().ToString(); in Compile()
911 executable->module_config().entry_computation_layout().parameter_count(); in Execute()
919 executable->module_config().entry_computation_layout().parameter_shape( in Execute()
Dexecutable.h360 return hlo_module_->config().entry_computation_layout().result_shape(); in result_shape()
Dlayout_assignment.h248 ComputationLayout* entry_computation_layout,
Dlayout_assignment.cc1241 ComputationLayout* entry_computation_layout, in LayoutAssignment() argument
1244 : entry_computation_layout_(entry_computation_layout), in LayoutAssignment()
1245 saved_entry_computation_layout_(*entry_computation_layout), in LayoutAssignment()
Dhlo_parser.cc926 std::optional<ComputationLayout> entry_computation_layout; in ParseHloModule() local
935 &entry_computation_layout}; in ParseHloModule()
953 if (entry_computation_layout.has_value()) { in ParseHloModule()
954 *config.mutable_entry_computation_layout() = *entry_computation_layout; in ParseHloModule()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dc_api_conversions.cc444 FromC(&se_config.entry_computation_layout.result_layout)); in ConfigWithLayout()
446 for (int i = 0; i < se_config.entry_computation_layout.parameter_count; ++i) { in ConfigWithLayout()
448 FromC(&se_config.entry_computation_layout.parameter_layouts[i]))); in ConfigWithLayout()
482 const auto& layout = config.entry_computation_layout(); in ToC()
484 &hlo_config.entry_computation_layout.result_layout); in ToC()
485 hlo_config.entry_computation_layout.parameter_layouts = in ToC()
490 &hlo_config.entry_computation_layout.parameter_layouts[i]); in ToC()
492 hlo_config.entry_computation_layout.parameter_count = in ToC()
531 for (auto i = 0; i < c_config->entry_computation_layout.parameter_count; in Destroy()
534 &c_config->entry_computation_layout.parameter_layouts[i]); in Destroy()
[all …]
Dc_api_decl.h316 XLA_ComputationLayout entry_computation_layout; member
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.h32 ComputationLayout* entry_computation_layout,
35 : LayoutAssignment(entry_computation_layout, channel_constraints), in LayoutAssignment() argument
Dcpu_layout_assignment_test.cc52 ComputationLayout* entry_computation_layout) { in AssignLayouts() argument
57 cpu::CpuLayoutAssignment layout_assignment(entry_computation_layout, in AssignLayouts()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.h33 ComputationLayout* entry_computation_layout,
36 : LayoutAssignment(entry_computation_layout, channel_constraints), in LayoutAssignment() argument
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc93 executable->executable()->module_config().entry_computation_layout(); in GetInputTuples()
304 executable->executable()->module_config().entry_computation_layout(); in RunExecutable()
355 .entry_computation_layout() in RunExecutable()
Dxrt_compile_ops.cc233 .entry_computation_layout() in Compute()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.cc458 module_config_0.entry_computation_layout() in RunAndCompareTwoModules()
462 ? module_config_0.entry_computation_layout() in RunAndCompareTwoModules()
470 module_config_1.entry_computation_layout() in RunAndCompareTwoModules()
474 ? module_config_1.entry_computation_layout() in RunAndCompareTwoModules()
Dtest_utils.cc747 module_config.entry_computation_layout() in MakeFakeArguments()
751 ? module_config.entry_computation_layout() in MakeFakeArguments()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc472 module->config().entry_computation_layout().result_shape()), in TPUExecute()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc123 executable_->module_config().entry_computation_layout(); in RunHelper()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.cc1186 cpu_executable_->module().entry_computation_layout(); in TfrtCpuExecutable()

12