HomeSort by relevance Sort by last modified time
    Searched refs:entry_computation_layout (Results 1 - 13 of 13) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment.h 30 explicit CpuLayoutAssignment(ComputationLayout* entry_computation_layout)
31 : LayoutAssignment(entry_computation_layout) {}
cpu_layout_assignment_test.cc 51 ComputationLayout* entry_computation_layout) {
52 cpu::CpuLayoutAssignment layout_assignment(entry_computation_layout);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment.h 30 explicit GpuLayoutAssignment(ComputationLayout* entry_computation_layout)
31 : LayoutAssignment(entry_computation_layout) {}
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_module_config.h 38 // accessing entry_computation_layout will CHECK-fail. The ctor accepting a
54 const ComputationLayout& entry_computation_layout() const { function in class:xla::HloModuleConfig
hlo_module.h 106 ComputationLayout entry_computation_layout() const { function in class:xla::HloModule
107 return config_.entry_computation_layout();
executable.h 152 return hlo_module_->config().entry_computation_layout().result_shape();
hlo_module.cc 285 module_config.entry_computation_layout().parameter_count());
288 module_config.entry_computation_layout().parameter_layout(i).shape();
298 module_config.entry_computation_layout().result_layout().shape();
layout_assignment.h 270 // entry_computation_layout is modified to populate a layout for the result in
281 ComputationLayout* entry_computation_layout,
layout_assignment_test.cc 55 ComputationLayout* entry_computation_layout) {
56 LayoutAssignment layout_assignment(entry_computation_layout);
490 ComputationLayout* entry_computation_layout)
491 : LayoutAssignment(entry_computation_layout) {}
    [all...]
service.cc 761 << module_config->entry_computation_layout().ToString();
    [all...]
layout_assignment.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
local_client.h 49 .entry_computation_layout()
local_client.cc 57 executable_->module_config().entry_computation_layout();

Completed in 546 milliseconds