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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_module_config.h 61 ComputationLayout* mutable_entry_computation_layout() { function in class:xla::HloModuleConfig
hlo_module.h 102 ComputationLayout* mutable_entry_computation_layout() { function in class:xla::HloModule
103 return config_.mutable_entry_computation_layout();
hlo_module.cc 365 module_config.mutable_entry_computation_layout();
service.cc 270 auto* computation_layout = config->mutable_entry_computation_layout();
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
hlo_test_base.h 178 module->mutable_entry_computation_layout()->parameter_count());
179 module->mutable_entry_computation_layout()
187 module->mutable_entry_computation_layout()
195 module->mutable_entry_computation_layout()
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
compiler.cc 51 hlo_module->mutable_entry_computation_layout());
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_compiler.cc 281 hlo_module->mutable_entry_computation_layout());
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_compiler.cc 295 module->mutable_entry_computation_layout());
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser.cc 307 module_->mutable_entry_computation_layout()
314 module_->mutable_entry_computation_layout()
    [all...]

Completed in 829 milliseconds