Searched refs:aot_options (Results 1 – 7 of 7) sorted by relevance
48 absl::optional<int> num_threads, const AotCompilationOptions* aot_options) { in CreateModuleConfig() argument123 if (aot_options != nullptr && in CreateModuleConfig()124 aot_options->fusion_config_collection() != FusionConfigCollection::kOff) { in CreateModuleConfig()126 aot_options->fusion_config_collection()); in CreateModuleConfig()127 *config->mutable_fusion_config() = aot_options->fusion_config(); in CreateModuleConfig()
40 const AotCompilationOptions* aot_options = nullptr);
192 const AotCompilationOptions* aot_options = nullptr);201 const AotCompilationOptions* aot_options = nullptr);
259 const AotCompilationOptions* aot_options) { in CreateModuleConfig() argument269 num_threads, aot_options); in CreateModuleConfig()276 const AotCompilationOptions* aot_options) { in CreateModuleConfig() argument282 aot_options); in CreateModuleConfig()
59 const AotCompilationOptions& aot_options) override;
156 const AotCompilationOptions& aot_options) { in CompileAheadOfTime() argument
805 const AotCompilationOptions& aot_options) { in CompileAheadOfTime() argument842 if (aot_options.PlatformId() != se::host::kHostPlatformId) { in CompileAheadOfTime()846 static_cast<const CpuAotCompilationOptions&>(aot_options); in CompileAheadOfTime()