| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_configuration_ops.cc | 44 if (!rmgr->Lookup(rmgr->default_container(), in GetTpuMeshStateInterface()
61 resource_manager->default_container(), resource_name); in DeleteIfExists()
78 rmgr->default_container(), tpu::kCompilationCacheResourceName, in CreateTpuCompilationCache()
135 ctx, rmgr->Create(rmgr->default_container(), in Compute()
265 OP_REQUIRES_OK(ctx, rmgr->Create(rmgr->default_container(), in Compute()
283 Status s = rmgr->Lookup(rmgr->default_container(), in Compute()
319 ctx, rmgr->Create(rmgr->default_container(), in Compute()
353 ctx, rmgr->Create(rmgr->default_container(), in Compute()
|
| D | tpu_pod_state.cc | 40 resource_manager->default_container(), resource_name); in DeleteIfExists()
125 if (!rmgr->Lookup(rmgr->default_container(), kTpuPodStateResourceName, in GetTPUPodState()
136 if (!rmgr->Lookup(rmgr->default_container(), kTpuPodStateResourceName, in HasTPUPodState()
185 return rmgr->Create(rmgr->default_container(), kTpuPodStateResourceName, in ConstructTpuPodState()
|
| D | tpu_handle_to_key_op.cc | 42 rm->default_container(), in Compute()
|
| D | tpu_compile_op_common.cc | 636 TF_RETURN_IF_ERROR(rm->Lookup(rm->default_container(), in ComputeInternal()
660 rm->default_container(), kCompilationCacheResourceName, &cache)); in ComputeInternal()
748 ctx->resource_manager()->default_container(), in ComputeInternal()
|
| D | tpu_execute_op.cc | 82 TF_RETURN_IF_ERROR(rmgr->Lookup(rmgr->default_container(), in GetComputationCacheEntry()
|
| /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 | 123 ResourceMgr::ResourceMgr(const string& default_container) in ResourceMgr() argument
124 : default_container_(default_container) {} in ResourceMgr()
297 container_ = rmgr_->default_container(); in Init()
|
| D | resource_mgr.h | 159 explicit ResourceMgr(const std::string& default_container);
163 const std::string& default_container() const { return default_container_; } in default_container() function
309 ? ctx->resource_manager()->default_container()
322 ? ctx->resource_manager()->default_container()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 160 rm->Lookup(rm->default_container(), in Compute()
188 rm->default_container(), in Compute()
249 rm->default_container(), in Compute()
|
| D | tpu_execute_op.cc | 66 TF_RETURN_IF_ERROR(rm->Lookup(rm->default_container(), in GetComputationCacheEntry()
322 rm->default_container(), tpu::kCompilationCacheResourceName, &cache)); in DoWork()
423 TF_RETURN_IF_ERROR(rm->Lookup(rm->default_container(), in DoWork()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 850 TF_ASSERT_OK(mgr0->Create(mgr0->default_container(), "var", resource0)); in TEST_F()
851 TF_ASSERT_OK(mgr1->Create(mgr1->default_container(), "var", resource1)); in TEST_F()
854 GetResourceHandle("var", mgr0->default_container(), device0_->name()); in TEST_F()
856 GetResourceHandle("var", mgr1->default_container(), device1_->name()); in TEST_F()
917 Status status = mgr->Create(mgr->default_container(), "my_gpu_var", resource); in TEST_F()
923 Tensor x2 = GetResourceHandle("my_gpu_var", mgr->default_container(), in TEST_F()
1047 Status status = mgr->Create(mgr->default_container(), "my_gpu_var", resource); in TEST_F()
1050 Tensor x = GetResourceHandle("my_gpu_var", mgr->default_container(), in TEST_F()
|
| D | device_mgr.cc | 137 dev->resource_manager()->default_container())); in ClearContainers()
|
| D | dynamic_device_mgr.cc | 107 pair.first->resource_manager()->default_container())); in ClearContainers()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.h | 127 container.empty() ? rm->default_container() : container; in AddResourceInput()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.cc | 97 rmgr->default_container(), "xla_cache", &cache, in GetCompilerIr()
|
| D | xla_compile_on_demand_op.cc | 123 rm->default_container(), "xla_cache", cache, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_compilation_cache.cc | 296 rm->default_container(), kXRTCompilationCacheResourceName, &cache, in GetOrCreateCompilationCache()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 642 const string& container = ctx->resource_manager()->default_container(); in MakeDataset()
906 const string& container = ctx->resource_manager()->default_container(); in MakeDataset()
|
| D | cache_dataset_ops.cc | 1030 const string& container = ctx->resource_manager()->default_container(); in MakeDataset()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 182 rm->default_container(), "xla_cache", &cache, in CompileToLocalExecutable()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | data_service_dataset_op.cc | 940 const std::string& container = ctx->resource_manager()->default_container(); in MakeDataset()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 109 rm->default_container(), "dummy", &dummy)); in Compile()
718 return rm->Create(rm->default_container(), "dummy", resource); in TEST_F()
|