| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | resource_mgr_test.cc | 346 ResourceMgr resource_mgr(""); in TEST() local
375 ResourceMgr resource_mgr(""); in TEST() local
405 ResourceMgr resource_mgr(""); in TEST() local
434 ResourceMgr resource_mgr(""); in TEST() local
450 ResourceMgr resource_mgr(""); in TEST() local
|
| H A D | dataset.h | 477 ResourceMgr* resource_mgr = nullptr; member
540 ResourceMgr* resource_mgr() { return params_.resource_mgr; } in resource_mgr() function
657 ResourceMgr* resource_mgr; member
675 const ResourceMgr* resource_mgr() const { return params_.resource_mgr; } in resource_mgr() function
|
| H A D | op_kernel.cc | 259 FunctionLibraryRuntime* flib, ResourceMgr* resource_mgr, in OpKernelConstruction()
1681 ResourceMgr* resource_mgr, in CreateOpKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/ |
| H A D | eval_utils_test.cc | 61 auto resource_mgr = std::make_unique<tensorflow::ResourceMgr>(); in TEST() local
97 auto resource_mgr = std::make_unique<tensorflow::ResourceMgr>(); in TEST() local
160 auto resource_mgr = std::make_unique<tensorflow::ResourceMgr>(); in TEST() local
|
| H A D | eval_utils.cc | 79 tensorflow::ResourceMgr *resource_mgr, TFOp op, in EvaluateOperation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | evaluation_utils.cc | 62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
|
| H A D | loop_optimizer.cc | 558 ResourceMgr* resource_mgr, in EvaluateBoolOpForConstantOperands()
600 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | reduce_dataset_op.cc | 67 ResourceMgr resource_mgr; in DoCompute() local
|
| H A D | iterator_ops.h | 116 ResourceMgr* resource_mgr() { return &resource_mgr_; } in resource_mgr() function
|
| H A D | multi_device_iterator_ops.cc | 165 ResourceMgr* resource_mgr() { return &resource_mgr_; } in resource_mgr() function in tensorflow::data::__anon7578bfd70111::MultiDeviceIterator
|
| H A D | iterator_ops.cc | 658 ResourceMgr resource_mgr; in DoCompute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | to_tf_record_op.cc | 82 ResourceMgr resource_mgr; in DoCompute() local
|
| H A D | io_ops.cc | 113 ResourceMgr resource_mgr; in WriteData() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_op_util.cc | 111 ResourceMgr* resource_mgr) { in CreateCompilationCacheKey()
|
| H A D | tpu_compile_op_common.cc | 244 ResourceMgr* resource_mgr = ctx->resource_manager(); in ComputeInternal() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | delegate_data.cc | 159 tensorflow::ResourceMgr* resource_mgr, in RegisterFunctionDefForSubgraphs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | partitioned_function_ops.cc | 239 ResourceMgr* resource_mgr = lib->device()->resource_manager(); in RunFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | captured_function.cc | 958 ResourceMgr* resource_mgr = lib_->device()->resource_manager(); in RunAsync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_ops_c_api.h | 93 tensorflow::ResourceMgr* resource_mgr; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 1638 ResourceMgr* resource_mgr = device_ptr->resource_manager(); in CreateContext() local
|