Searched refs:default_container (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_op_kernel_test.cc | 147 mgr_.Lookup<StubResource>(mgr_.default_container(), key, &resource)); in TEST_F()
155 mgr_.Lookup<StubResource>(mgr_.default_container(), key, &resource); in TEST_F()
168 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name, in TEST_F()
176 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name, in TEST_F()
|
| D | resource_mgr.cc | 120 ResourceMgr::ResourceMgr(const string& default_container) in ResourceMgr() argument
121 : default_container_(default_container) {} in ResourceMgr()
294 container_ = rmgr_->default_container(); in Init()
|
| D | resource_mgr.h | 159 explicit ResourceMgr(const string& default_container);
163 const string& default_container() const { return default_container_; } in default_container() function
306 container.empty() ? ctx->resource_manager()->default_container()
316 container.empty() ? ctx->resource_manager()->default_container()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.h | 126 container.empty() ? rm->default_container() : container; in AddResourceInput()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_mgr.cc | 132 dev->resource_manager()->default_container())); in ClearContainers()
|
| D | process_function_library_runtime_test.cc | 736 Status status = mgr->Create(mgr->default_container(), "my_gpu_var", resource); in TEST_F()
743 Tensor x2 = GetResourceHandle("my_gpu_var", mgr->default_container(), in TEST_F()
867 Status status = mgr->Create(mgr->default_container(), "my_gpu_var", resource); in TEST_F()
870 Tensor x = GetResourceHandle("my_gpu_var", mgr->default_container(), in TEST_F()
|
| D | dynamic_device_mgr.cc | 102 pair.first->resource_manager()->default_container())); in ClearContainers()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_compilation_cache.cc | 296 rm->default_container(), kXRTCompilationCacheResourceName, &cache, in GetOrCreateCompilationCache()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 177 rm->default_container(), "xla_cache", &cache, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 220 rm->default_container(), in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 288 rm->default_container(), "xla_cache", &cache, in CompileToLocalExecutable()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 109 rm->default_container(), "dummy", &dummy)); in Compile()
701 return rm->Create(rm->default_container(), "dummy", resource); in TEST_F()
|