Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.h32 ComputationLayout* entry_computation_layout, in CpuLayoutAssignment() argument
36 : LayoutAssignment(entry_computation_layout, in CpuLayoutAssignment()
Dcpu_layout_assignment_test.cc52 ComputationLayout* entry_computation_layout) { in AssignLayouts() argument
58 entry_computation_layout, LayoutAssignment::InstructionCanChangeLayout, in AssignLayouts()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.h32 explicit GpuLayoutAssignment(ComputationLayout* entry_computation_layout, in GpuLayoutAssignment() argument
36 : LayoutAssignment(entry_computation_layout, in GpuLayoutAssignment()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc54 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()
1151 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1167 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1194 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1226 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1253 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
1288 ComputationLayout computation_layout = m->entry_computation_layout(); in TEST_F()
[all …]
Dhlo_module.h131 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
132 return config_.entry_computation_layout(); in entry_computation_layout()
Dhlo_module.cc237 entry_computation_layout().ComputeProgramShape().ToProto(); in ToProto()
284 module_config.entry_computation_layout().parameter_count()); in CreateFromProto()
287 module_config.entry_computation_layout().parameter_layout(i).shape(); in CreateFromProto()
297 module_config.entry_computation_layout().result_layout().shape(); in CreateFromProto()
Dhlo_module_config.h59 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
Dexecutable.h174 return hlo_module_->config().entry_computation_layout().result_shape(); in result_shape()
Dlocal_service.cc172 << module_config->entry_computation_layout().ToString(); in CompileExecutable()
Dservice.cc666 << module_config->entry_computation_layout().ToString(); in ExecuteGraphParallel()
840 << module_config->entry_computation_layout().ToString(); in Compile()
872 executable->module_config().entry_computation_layout().parameter_count(); in Execute()
880 executable->module_config().entry_computation_layout().parameter_shape( in Execute()
Dlayout_assignment.h302 ComputationLayout* entry_computation_layout,
Dlayout_assignment.cc968 ComputationLayout* entry_computation_layout, in LayoutAssignment() argument
972 : entry_computation_layout_(entry_computation_layout), in LayoutAssignment()
974 saved_entry_computation_layout_(*entry_computation_layout), in LayoutAssignment()
Dhlo_verifier.cc944 const auto& layout = module.entry_computation_layout(); in VerifyEntryComputationLayout()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_compile_ops.cc190 .entry_computation_layout() in Compute()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc58 executable_->module_config().entry_computation_layout(); in ValidateExecutionOptions()