| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_mgr.cc | 81 Status ResourceMgr::InsertDebugTypeName(uint64 hash_code, in InsertDebugTypeName()
91 const char* ResourceMgr::DebugTypeName(uint64 hash_code) const { in DebugTypeName()
100 ResourceMgr::ResourceMgr() : default_container_("localhost") {} in ResourceMgr() function in tensorflow::ResourceMgr
102 ResourceMgr::ResourceMgr(const string& default_container) in ResourceMgr() function in tensorflow::ResourceMgr
105 ResourceMgr::~ResourceMgr() { Clear(); } in ~ResourceMgr()
107 void ResourceMgr::Clear() { in Clear()
118 string ResourceMgr::DebugString() const { in DebugString()
149 Status ResourceMgr::DoCreate(const string& container, TypeIndex type, in DoCreate()
164 Status ResourceMgr::DoLookup(const string& container, TypeIndex type, in DoLookup()
183 Status ResourceMgr::DoDelete(const string& container, uint64 type_hash_code, in DoDelete()
[all …]
|
| D | resource_mgr_test.cc | 53 string Find(const ResourceMgr& rm, const string& container, in Find()
63 string LookupOrCreate(ResourceMgr* rm, const string& container, in LookupOrCreate()
81 Status FindErr(const ResourceMgr& rm, const string& container, in FindErr()
90 ResourceMgr rm; in TEST()
129 ResourceMgr rm; in TEST()
141 ResourceMgr rm; in TEST()
168 ResourceMgr rmgr; in ComputePolicy()
253 ResourceMgr resource_mgr(""); in TEST()
283 ResourceMgr resource_mgr(""); in TEST()
293 ResourceMgr other_resource_mgr(""); in TEST()
[all …]
|
| D | resource_mgr.h | 112 class ResourceMgr {
114 ResourceMgr();
115 explicit ResourceMgr(const string& default_container);
116 ~ResourceMgr();
231 TF_DISALLOW_COPY_AND_ASSIGN(ResourceMgr);
318 Status Init(ResourceMgr* rmgr, const NodeDef& ndef,
320 Status Init(ResourceMgr* rmgr, const NodeDef& ndef) { in Init()
328 ResourceMgr* resource_manager() const { return rmgr_; } in resource_manager()
339 ResourceMgr* rmgr_ = nullptr;
432 Status ResourceMgr::Create(const string& container, const string& name, in Create()
[all …]
|
| D | device_base.h | 49 class ResourceMgr; variable
190 Allocator* GetStepAllocator(AllocatorAttributes attr, ResourceMgr*) { in GetStepAllocator() argument
|
| D | resource_op_kernel.h | 76 ResourceMgr* mgr = context->resource_manager(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.h | 134 static Status Lookup(ResourceMgr* rm, int64 key,
138 static Status DeleteFromResourceManager(ResourceMgr* rm, int64 key);
142 static Status ReleaseAllAllocations(ResourceMgr* rm);
146 Status Intern(ResourceMgr* rm, int64* key);
|
| D | xrt_state.cc | 272 /*static*/ Status XRTTupleAllocation::Lookup(ResourceMgr* rm, int64 key, in Lookup()
279 /*static*/ Status XRTTupleAllocation::DeleteFromResourceManager(ResourceMgr* rm, in DeleteFromResourceManager()
285 /* static */ Status XRTTupleAllocation::ReleaseAllAllocations(ResourceMgr* rm) { in ReleaseAllAllocations()
486 Status XRTTupleAllocation::Intern(ResourceMgr* rm, int64* key) { in Intern()
|
| D | xrt_device.h | 32 static Status GetResourceManager(OpKernelContext* ctx, ResourceMgr** rm);
|
| D | xrt_device.cc | 28 OpKernelContext* ctx, ResourceMgr** rm) { in GetResourceManager()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 72 xla::Shape* shape, ResourceMgr* rm) { in ParseTupleNode()
125 int* device_ordinal, ResourceMgr* rm) { in ParseTupleTree()
187 ResourceMgr* rm; in Compute()
286 ResourceMgr* rm; in Compute()
342 ResourceMgr* rm; in Compute()
409 ResourceMgr* rm; in Compute()
473 ResourceMgr* rm; in Compute()
529 ResourceMgr* rm; in Compute()
624 ResourceMgr* rm; in Compute()
657 ResourceMgr* rm; in Compute()
[all …]
|
| D | xrt_compile_ops.cc | 144 ResourceMgr* rm; in Compute()
220 ResourceMgr* rm; in Compute()
|
| D | xrt_execute_op.cc | 68 Status GetComputationInputs(OpKernelContext* context, ResourceMgr* rm, in GetComputationInputs()
141 ResourceMgr* rm; in DoWork()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device.h | 175 virtual ResourceMgr* resource_manager() { return rmgr_; } in resource_manager()
218 ResourceMgr* rmgr_ = nullptr;
|
| D | device.cc | 30 rmgr_ = new ResourceMgr(parsed_name_.job); in Device()
|
| D | renamed_device.h | 125 ResourceMgr* resource_manager() override { in resource_manager()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | variable_ops.cc | 94 ResourceMgr* rm = context->resource_manager(); in Compute()
146 ResourceMgr* rm = context->resource_manager(); in Compute()
|
| D | stack.cc | 140 ResourceMgr* rm = ctx->resource_manager(); in GetStack()
184 ResourceMgr* rm = ctx->resource_manager(); in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.h | 56 DeviceBase* cpu_device, ResourceMgr* resource_mgr,
|
| D | loop_optimizer.h | 70 std::unique_ptr<ResourceMgr> resource_mgr_;
|
| D | evaluation_utils.cc | 83 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | get_serialized_resource_op_test.cc | 54 ResourceMgr* rm = device->resource_manager(); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 291 ResourceMgr resource_mgr;
436 ResourceMgr* mgr = context->resource_manager(); in Compute()
516 ResourceMgr* mgr = context->resource_manager(); in Compute()
583 std::unique_ptr<ResourceMgr> resource_mgr = in ComputeAsync()
584 absl::make_unique<ResourceMgr>(); in ComputeAsync()
674 std::unique_ptr<ResourceMgr> resource_mgr = in ComputeAsync()
675 absl::make_unique<ResourceMgr>(); in ComputeAsync()
|
| D | captured_function.h | 33 class ResourceMgr; variable
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/test_kernels/ |
| D | bigtable_test_client_op.cc | 41 ResourceMgr* mgr = ctx->resource_manager(); in Compute()
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
| D | bigtable_kernels.cc | 69 ResourceMgr* mgr = ctx->resource_manager(); in Compute()
140 ResourceMgr* mgr = ctx->resource_manager(); in Compute()
|