| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_mgr.cc | 80 Status ResourceMgr::InsertDebugTypeName(uint64 hash_code, in InsertDebugTypeName()
90 const char* ResourceMgr::DebugTypeName(uint64 hash_code) const { in DebugTypeName()
99 ResourceMgr::ResourceAndName::ResourceAndName() in ResourceAndName()
102 ResourceMgr::ResourceAndName::ResourceAndName(ResourceBase* resource, in ResourceAndName()
106 ResourceMgr::ResourceAndName::ResourceAndName( in ResourceAndName()
112 ResourceMgr::ResourceAndName::~ResourceAndName() {} in ~ResourceAndName()
114 ResourceMgr::ResourceAndName& ResourceMgr::ResourceAndName::operator=( in operator =()
121 ResourceMgr::ResourceMgr() : default_container_("localhost") {} in ResourceMgr() function in tensorflow::ResourceMgr
123 ResourceMgr::ResourceMgr(const string& default_container) in ResourceMgr() function in tensorflow::ResourceMgr
126 ResourceMgr::~ResourceMgr() { Clear(); } in ~ResourceMgr()
[all …]
|
| D | resource_mgr_test.cc | 57 string Find(const ResourceMgr& rm, const string& container, in Find()
67 string LookupOrCreate(ResourceMgr* rm, const string& container, in LookupOrCreate()
85 Status FindErr(const ResourceMgr& rm, const string& container, in FindErr()
94 ResourceMgr rm; in TEST()
133 ResourceMgr rm; in TEST()
145 ResourceMgr rm; in TEST()
172 ResourceMgr rmgr; in ComputePolicy()
260 ResourceMgr resource_mgr(""); in TEST()
289 ResourceMgr resource_mgr(""); in TEST()
299 ResourceMgr other_resource_mgr(""); in TEST()
[all …]
|
| D | resource_mgr.h | 133 Status Create(ResourceMgr* rm, const std::string& name,
137 Status Delete(ResourceMgr* rm, const std::string& name) TF_MUST_USE_RESULT;
140 Status Lookup(ResourceMgr* rm, const std::string& name,
144 Status LookupOrCreate(ResourceMgr* rm, const std::string& name, T** resource,
156 class ResourceMgr {
158 ResourceMgr();
159 explicit ResourceMgr(const std::string& default_container);
160 ~ResourceMgr();
291 TF_DISALLOW_COPY_AND_ASSIGN(ResourceMgr);
413 Status Init(ResourceMgr* rmgr, const NodeDef& ndef,
[all …]
|
| D | device.h | 159 virtual ResourceMgr* resource_manager() { return rmgr_; } in resource_manager()
198 ResourceMgr* rmgr_ = nullptr;
|
| D | device_base.h | 48 class ResourceMgr; variable
189 Allocator* GetStepAllocator(AllocatorAttributes attr, ResourceMgr*) { in GetStepAllocator() argument
|
| D | device.cc | 30 rmgr_ = new ResourceMgr(parsed_name_.job); in Device()
|
| D | op_kernel.h | 77 class ResourceMgr; variable
261 ResourceMgr* resource_mgr,
352 ResourceMgr* resource_manager() const { return resource_mgr_; } in resource_manager()
380 ResourceMgr* const resource_mgr_;
640 ResourceMgr* resource_manager = nullptr;
1143 ResourceMgr* resource_manager() const { return params_->resource_manager; } in resource_manager()
1393 ResourceMgr* resource_mgr,
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_configuration.cc | 22 ResourceMgr* GetGlobalResourceMgr() { in GetGlobalResourceMgr()
23 static ResourceMgr* const rmgr = new ResourceMgr(); in GetGlobalResourceMgr()
34 ResourceMgr* GetTPUConfigResourceMgr() { in GetTPUConfigResourceMgr()
|
| D | tpu_configuration.h | 26 ResourceMgr* GetTPUConfigResourceMgr();
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_device.cc | 38 ResourceMgr* GetResourceMgr(const std::string& platform_name) { in GetResourceMgr()
42 it = resource_managers_.emplace(platform_name, new ResourceMgr()).first; in GetResourceMgr()
49 std::map<std::string, ResourceMgr*> resource_managers_;
55 OpKernelContext* ctx, ResourceMgr** rm) { in GetResourceManager()
65 ResourceMgr* rm; in GetOrCreateCompilationCache()
|
| D | xrt_tpu_device.h | 32 static Status GetResourceManager(OpKernelContext* ctx, ResourceMgr** rm);
|
| D | xrt_tpu_device.cc | 27 ResourceMgr** rm) { in GetResourceManager()
|
| D | xrt_device.h | 33 static Status GetResourceManager(OpKernelContext* ctx, ResourceMgr** rm);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_pod_state.h | 48 Status GetTPUPodState(const ResourceMgr* rmgr, TpuPodState** pod_state);
52 bool HasTPUPodState(const ResourceMgr* rmgr);
56 ResourceMgr* rmgr, const std::vector<int32_t>& num_devices_per_host,
|
| D | tpu_pod_state.cc | 36 Status DeleteIfExists(ResourceMgr* resource_manager, in DeleteIfExists()
54 ConstructCacheService(ResourceMgr* rmgr, int serving_port, in ConstructCacheService()
121 Status GetTPUPodState(const ResourceMgr* rmgr, TpuPodState** pod_state) { in GetTPUPodState()
134 bool HasTPUPodState(const ResourceMgr* rmgr) { in HasTPUPodState()
146 ResourceMgr* rmgr, const std::vector<int32_t>& num_devices_per_host, in ConstructTpuPodState()
|
| D | tpu_configuration_ops.cc | 42 Status GetTpuMeshStateInterface(const ResourceMgr* rmgr, in GetTpuMeshStateInterface()
57 Status DeleteIfExists(ResourceMgr* resource_manager, in DeleteIfExists()
76 ResourceMgr* rmgr, tpu::TpuCompilationCacheInterface** compilation_cache) { in CreateTpuCompilationCache()
117 ResourceMgr* rmgr = GetTPUConfigResourceMgr(); in Compute()
|
| D | tpu_handle_to_key_op.cc | 39 ResourceMgr* rm = GetTPUConfigResourceMgr(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 76 xla::Shape* shape, ResourceMgr* rm) { in ParseTupleNode()
129 int* device_ordinal, ResourceMgr* rm) { in ParseTupleTree()
192 ResourceMgr* rm; in Compute()
232 ResourceMgr* rm; in Compute()
336 ResourceMgr* rm; in Compute()
390 ResourceMgr* rm; in Compute()
444 ResourceMgr* rm; in Compute()
503 ResourceMgr* rm; in Compute()
556 ResourceMgr* rm; in Compute()
649 ResourceMgr* rm; in Compute()
[all …]
|
| D | tpu_compile_ops.cc | 157 ResourceMgr* rm = GetTPUConfigResourceMgr(); in Compute()
243 ResourceMgr* rm = GetTPUConfigResourceMgr(); in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.h | 55 DeviceBase* cpu_device, ResourceMgr* resource_mgr,
|
| D | loop_optimizer.h | 74 std::unique_ptr<ResourceMgr> resource_mgr_;
|
| D | evaluation_utils.cc | 62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | variable_ops.cc | 108 ResourceMgr* rm = context->resource_manager(); in Compute()
161 ResourceMgr* rm = context->resource_manager(); in Compute()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.h | 122 ResourceMgr resource_mgr_;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 106 ResourceMgr* resource_mgr() { return &resource_mgr_; } in resource_mgr()
119 ResourceMgr resource_mgr_;
|