Home
last modified time | relevance | path

Searched refs:CompilationEnvironments (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcompilation_environments.h44 class CompilationEnvironments {
46 CompilationEnvironments() = default;
47 CompilationEnvironments(const CompilationEnvironments& rhs) { *this = rhs; } in CompilationEnvironments() function
48 CompilationEnvironments& operator=(const CompilationEnvironments& rhs);
49 ~CompilationEnvironments() = default;
110 CompilationEnvironments::CreateDefaultEnv() = delete;
113 const T& CompilationEnvironments::GetEnv() { in GetEnv()
Dcompilation_environments_test.cc32 CompilationEnvironments::CreateDefaultEnv<test::TestCompilationEnvironment1>() { in CreateDefaultEnv()
39 CompilationEnvironments::CreateDefaultEnv<test::TestCompilationEnvironment2>() { in CreateDefaultEnv()
46 CompilationEnvironments::CreateDefaultEnv<test::TestCompilationEnvironment3>() { in CreateDefaultEnv()
58 CompilationEnvironments envs; in TEST_F()
64 CompilationEnvironments envs; in TEST_F()
72 CompilationEnvironments envs; in TEST_F()
80 auto envs = std::make_unique<CompilationEnvironments>(); in TEST_F()
89 auto envs_copy = std::make_unique<CompilationEnvironments>(*envs); in TEST_F()
100 auto envs1 = std::make_unique<CompilationEnvironments>(); in TEST_F()
110 auto envs2 = std::make_unique<CompilationEnvironments>(); in TEST_F()
Dcompilation_environments.cc113 void CompilationEnvironments::DefaultEnvCreated(std::string_view env_type) { in DefaultEnvCreated()
117 void CompilationEnvironments::DefaultEnvCreatedByCompilationEnvironments( in DefaultEnvCreatedByCompilationEnvironments()
123 void CompilationEnvironments::EnvAdded(std::string_view env_type) { in EnvAdded()
127 CompilationEnvironments& CompilationEnvironments::operator=( in operator =()
128 const CompilationEnvironments& rhs) { in operator =()
138 void CompilationEnvironments::AddEnv( in AddEnv()
Dhlo_module.h516 CompilationEnvironments& comp_envs() const { return *comp_envs_; } in comp_envs()
522 std::unique_ptr<CompilationEnvironments> comp_envs);
601 std::unique_ptr<CompilationEnvironments> comp_envs_ =
602 std::make_unique<CompilationEnvironments>();
Dhlo_module.cc58 : HloModule(name, config, std::make_unique<CompilationEnvironments>()) {} in HloModule()
873 std::make_unique<CompilationEnvironments>(*comp_envs_))); in Clone()
972 std::unique_ptr<CompilationEnvironments> comp_envs) in HloModule()
Dhlo_module_test.cc43 CompilationEnvironments::CreateDefaultEnv<test::TestCompilationEnvironment1>() { in CreateDefaultEnv()