Searched defs:hlo_module_ (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_context.h37 : hlo_module_(hlo_module),
46 const HloModule& hlo_module() const { return *hlo_module_; }
57 const HloModule* hlo_module_; member in class:xla::gpu::IrEmitterContext
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dexecutable.h51 : hlo_module_(std::move(hlo_module)),
137 const HloModule& module() const { return *hlo_module_; }
139 const bool has_module() const { return hlo_module_ != nullptr; }
141 const HloModuleConfig& module_config() const { return hlo_module_->config(); }
146 return hlo_module_->entry_computation_handle();
152 return hlo_module_->config().entry_computation_layout().result_shape();
176 const std::unique_ptr<const HloModule> hlo_module_; member in class:xla::Executable

Completed in 155 milliseconds