| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.cc | 404 FromC(&se_config.entry_computation_layout.result_layout)); in ConfigWithLayout()
406 for (int i = 0; i < se_config.entry_computation_layout.parameter_count; ++i) { in ConfigWithLayout()
408 FromC(&se_config.entry_computation_layout.parameter_layouts[i]))); in ConfigWithLayout()
437 const auto& layout = config.entry_computation_layout(); in ToC()
439 &hlo_config.entry_computation_layout.result_layout); in ToC()
440 hlo_config.entry_computation_layout.parameter_layouts = in ToC()
445 &hlo_config.entry_computation_layout.parameter_layouts[i]); in ToC()
447 hlo_config.entry_computation_layout.parameter_count = in ToC()
476 for (auto i = 0; i < c_config->entry_computation_layout.parameter_count; in Free()
479 &c_config->entry_computation_layout.parameter_layouts[i]); in Free()
[all …]
|
| D | c_api_decl.h | 286 XLA_ComputationLayout entry_computation_layout; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.h | 32 ComputationLayout* entry_computation_layout, in CpuLayoutAssignment() argument
36 : LayoutAssignment(entry_computation_layout, in CpuLayoutAssignment()
|
| D | cpu_layout_assignment_test.cc | 52 ComputationLayout* entry_computation_layout) { in AssignLayouts() argument
58 entry_computation_layout, LayoutAssignment::InstructionCanChangeLayout, in AssignLayouts()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.h | 32 explicit GpuLayoutAssignment(ComputationLayout* entry_computation_layout, in GpuLayoutAssignment() argument
36 : LayoutAssignment(entry_computation_layout, in GpuLayoutAssignment()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_config.cc | 36 HloModuleConfig::HloModuleConfig(ComputationLayout entry_computation_layout) in HloModuleConfig() argument
37 : entry_computation_layout_(std::move(entry_computation_layout)) {} in HloModuleConfig()
|
| D | hlo_module_config.h | 77 explicit HloModuleConfig(ComputationLayout entry_computation_layout);
95 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
|
| D | layout_assignment_test.cc | 54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts() argument
57 entry_computation_layout, LayoutAssignment::InstructionCanChangeLayout, in AssignLayouts()
517 ComputationLayout* entry_computation_layout) in OperandsMustBeTheSameLayoutAssignment() argument
518 : LayoutAssignment(entry_computation_layout) {} in OperandsMustBeTheSameLayoutAssignment()
1145 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1161 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1188 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1220 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1247 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1282 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
[all …]
|
| D | hlo_module.cc | 285 entry_computation_layout().ComputeProgramShape().ToProto(); in ToProto()
342 module_config.entry_computation_layout().parameter_count()); in CreateFromProto()
345 module_config.entry_computation_layout().parameter_layout(i).shape(); in CreateFromProto()
355 module_config.entry_computation_layout().result_layout().shape(); in CreateFromProto()
787 uint64 result = entry_computation_layout().Hash(); in Hash()
|
| D | hlo_module.h | 148 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
149 return config_.entry_computation_layout(); in entry_computation_layout()
|
| D | local_service.cc | 154 << module_config->entry_computation_layout().ToString(); in CompileExecutables()
|
| D | service.cc | 618 << module_config->entry_computation_layout().ToString(); in ExecuteGraphParallel()
816 << module_config->entry_computation_layout().ToString(); in Compile()
848 executable->module_config().entry_computation_layout().parameter_count(); in Execute()
856 executable->module_config().entry_computation_layout().parameter_shape( in Execute()
|
| D | executable.h | 362 return hlo_module_->config().entry_computation_layout().result_shape(); in result_shape()
|
| D | layout_assignment.h | 304 ComputationLayout* entry_computation_layout,
|
| D | layout_assignment.cc | 1025 ComputationLayout* entry_computation_layout, in LayoutAssignment() argument
1029 : entry_computation_layout_(entry_computation_layout), in LayoutAssignment()
1031 saved_entry_computation_layout_(*entry_computation_layout), in LayoutAssignment()
|
| D | hlo_verifier.cc | 1277 const auto& layout = module.entry_computation_layout(); in VerifyEntryComputationLayout()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 93 executable->executable()->module_config().entry_computation_layout(); in GetInputTuples()
303 executable->executable()->module_config().entry_computation_layout(); in RunExecutable()
351 .entry_computation_layout() in RunExecutable()
|
| D | xrt_compile_ops.cc | 232 .entry_computation_layout() in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 690 module_config.entry_computation_layout() in MakeFakeArguments()
694 ? module_config.entry_computation_layout() in MakeFakeArguments()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 468 module->config().entry_computation_layout().result_shape()), in TPUExecute()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 122 executable_->module_config().entry_computation_layout(); in RunHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3700 const auto& old_entry_layout = module->entry_computation_layout(); in Run()
|