Home
last modified time | relevance | path

Searched defs:resource_manager (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dresource_mgr.h457 ResourceMgr* resource_manager() const { return rmgr_; } in resource_manager() function
565 ResourceDeleter(ResourceHandle handle, ResourceMgr* resource_manager) in ResourceDeleter()
616 ResourceMgr* resource_manager; // not owned member
Ddevice.h159 virtual ResourceMgr* resource_manager() { return rmgr_; } in resource_manager() function
Dop_kernel.h319 ResourceMgr* resource_manager() const { return resource_mgr_; } in resource_manager() function
613 ResourceMgr* resource_manager = nullptr; member
1099 ResourceMgr* resource_manager() const { return params_->resource_manager; } in resource_manager() function
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_v2_ops.cc163 ResourceMgr* resource_manager = context->resource_manager(); in Compute() local
230 ResourceMgr* resource_manager = context->resource_manager(); in Compute() local
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dop_kernel_runner.h91 tensorflow::ResourceMgr* resource_manager() const { in resource_manager() function
Dkernel_fallback_execute_compat.cc818 ResourceMgr* resource_manager() override { in resource_manager() function in tensorflow::tfd::__anone1369dfd0711::DeviceWithCustomAllocator
/external/tensorflow/tensorflow/core/tfrt/fallback/
Dop_kernel_runner.h72 tensorflow::ResourceMgr* resource_manager() const { in resource_manager() function
/external/tensorflow/tensorflow/core/distributed_runtime/
Dremote_device.cc45 ResourceMgr* resource_manager() override { in resource_manager() function in tensorflow::RemoteDevice
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h143 ResourceMgr* resource_manager() override { in resource_manager() function
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_pod_state.cc37 Status DeleteIfExists(ResourceMgr* resource_manager, in DeleteIfExists()
Dtpu_configuration_ops.cc78 Status DeleteIfExists(ResourceMgr* resource_manager, in DeleteIfExists()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_tpu_kernels.cc54 Status DeleteIfExists(ResourceMgr* resource_manager, in DeleteIfExists()