Searched refs:hlo_module_ (Results 1 – 3 of 3) sorted by relevance
67 : hlo_module_(std::move(hlo_module)), in Executable()165 HloModule& module() const { return *hlo_module_; } in module()167 const bool has_module() const { return hlo_module_ != nullptr; } in has_module()169 const HloModuleConfig& module_config() const { return hlo_module_->config(); } in module_config()174 return hlo_module_->config().entry_computation_layout().result_shape(); in result_shape()197 const std::unique_ptr<HloModule> hlo_module_; variable
37 : hlo_module_(hlo_module), in IrEmitterContext()46 const HloModule& hlo_module() const { return *hlo_module_; } in hlo_module()57 const HloModule* hlo_module_;
98 sub_streams, hlo_module_->entry_computation()); in ExecuteThunks()106 ScopedAnnotation top_level_annotation(hlo_module_->name(), "XLA GPU module"); in ExecuteThunks()288 HloInstruction* root = hlo_module_->entry_computation()->root_instruction(); in ExecuteOnStream()