Searched refs:default_container (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_op_kernel_test.cc | 144 mgr_.Lookup<StubResource>(mgr_.default_container(), key, &resource)); in TEST_F()
152 mgr_.Lookup<StubResource>(mgr_.default_container(), key, &resource); in TEST_F()
165 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name, in TEST_F()
173 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name, in TEST_F()
|
| D | resource_mgr.cc | 44 actual_container = ctx->resource_manager()->default_container(); in MakeResourceHandle()
102 ResourceMgr::ResourceMgr(const string& default_container) in ResourceMgr() argument
103 : default_container_(default_container) {} in ResourceMgr()
271 container_ = rmgr_->default_container(); in Init()
|
| D | resource_mgr.h | 115 explicit ResourceMgr(const string& default_container);
119 const string& default_container() const { return default_container_; } in default_container() function
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 165 rm->default_container(), kXRTCompilationCacheResourceName, in Compute()
226 rm->default_container(), in Compute()
|
| D | xrt_execute_op.cc | 162 rm->default_container(), kXRTCompilationCacheResourceName, &cache)); in DoWork()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_mgr.cc | 117 dev->resource_manager()->default_container())); in ClearContainers()
|
| D | process_function_library_runtime_test.cc | 671 Status status = mgr->Create(mgr->default_container(), "my_gpu_var", resource); in TestResourceOutputAndUse()
676 Tensor x2 = GetResourceHandle("my_gpu_var", mgr->default_container(), in TestResourceOutputAndUse()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 173 rm->default_container(), "xla_cache", &cache, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.h | 144 container == "" ? rm->default_container() : container; in AddResourceInput()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 270 rm->default_container(), "xla_cache", &cache, in CompileToLocalExecutable()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 101 rm->default_container(), "dummy", &dummy)); in Compile()
647 return rm->Create(rm->default_container(), "dummy", resource); in TEST_F()
|