Home
last modified time | relevance | path

Searched defs:resource_mgr (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/transforms/utils/
Deval_utils_test.cc61 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
Deval_utils.cc79 tensorflow::ResourceMgr *resource_mgr, TFOp op, in EvaluateOperation()
/external/tensorflow/tensorflow/core/framework/
Dresource_mgr_test.cc346 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
Ddataset.h477 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
Dop_kernel.cc259 FunctionLibraryRuntime* flib, ResourceMgr* resource_mgr, in OpKernelConstruction()
1681 ResourceMgr* resource_mgr, in CreateOpKernel()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
Dloop_optimizer.cc558 ResourceMgr* resource_mgr, in EvaluateBoolOpForConstantOperands()
600 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout()
/external/tensorflow/tensorflow/core/kernels/data/
Dreduce_dataset_op.cc67 ResourceMgr resource_mgr; in DoCompute() local
Diterator_ops.h116 ResourceMgr* resource_mgr() { return &resource_mgr_; } in resource_mgr() function
Dmulti_device_iterator_ops.cc165 ResourceMgr* resource_mgr() { return &resource_mgr_; } in resource_mgr() function in tensorflow::data::__anon2efe57860111::MultiDeviceIterator
Diterator_ops.cc657 ResourceMgr resource_mgr; in DoCompute() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc82 ResourceMgr resource_mgr; in DoCompute() local
Dio_ops.cc113 ResourceMgr resource_mgr; in WriteData() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_op_util.cc111 ResourceMgr* resource_mgr) { in CreateCompilationCacheKey()
Dtpu_compile_op_common.cc244 ResourceMgr* resource_mgr = ctx->resource_manager(); in ComputeInternal() local
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_data.cc159 tensorflow::ResourceMgr* resource_mgr, in RegisterFunctionDefForSubgraphs()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc239 ResourceMgr* resource_mgr = lib->device()->resource_manager(); in RunFunction() local
/external/tensorflow/tensorflow/core/data/
Dcaptured_function.cc958 ResourceMgr* resource_mgr = lib_->device()->resource_manager(); in RunAsync() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h93 tensorflow::ResourceMgr* resource_mgr; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1638 ResourceMgr* resource_mgr = device_ptr->resource_manager(); in CreateContext() local