Home
last modified time | relevance | path

Searched refs:resource_mgr (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dresource_mgr_test.cc260 ResourceMgr resource_mgr(""); in TEST() local
262 params.resource_manager = &resource_mgr; in TEST()
289 ResourceMgr resource_mgr(""); in TEST() local
291 params.resource_manager = &resource_mgr; in TEST()
318 ResourceMgr resource_mgr(""); in TEST() local
320 params.resource_manager = &resource_mgr; in TEST()
334 ResourceMgr resource_mgr(""); in TEST() local
336 params.resource_manager = &resource_mgr; in TEST()
Ddataset.h347 resource_mgr(ctx->resource_mgr()), in Params()
408 ResourceMgr* resource_mgr = nullptr; member
458 ResourceMgr* resource_mgr() { return params_.resource_mgr; } in resource_mgr() function
DBUILD81 "resource_mgr.h",
225 "resource_mgr.h",
283 "resource_mgr.cc",
412 "resource_mgr.cc",
413 "resource_mgr.h",
Dop_kernel.cc230 FunctionLibraryRuntime* flib, ResourceMgr* resource_mgr, in OpKernelConstruction() argument
239 resource_mgr_(resource_mgr), in OpKernelConstruction()
1613 ResourceMgr* resource_mgr, in CreateOpKernel() argument
1660 resource_mgr, props, input_memory_types, in CreateOpKernel()
Dop_kernel.h261 ResourceMgr* resource_mgr,
1393 ResourceMgr* resource_mgr,
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc80 ResourceMgr resource_mgr; in DoCompute() local
81 params.resource_mgr = &resource_mgr; in DoCompute()
Dio_ops.cc80 ResourceMgr resource_mgr; in WriteData() local
81 params.resource_mgr = &resource_mgr; in WriteData()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode() argument
80 params.resource_manager = resource_mgr; in EvaluateNode()
Devaluation_utils.h55 DeviceBase* cpu_device, ResourceMgr* resource_mgr,
Dloop_optimizer.cc558 ResourceMgr* resource_mgr, in EvaluateBoolOpForConstantOperands() argument
576 EvaluateNode(op_node, inputs, cpu_device, resource_mgr, &outputs)); in EvaluateBoolOpForConstantOperands()
600 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout() argument
703 *switch_ctrl_node, *operand_0, *operand_1, cpu_device, resource_mgr, in CheckForDeadFanout()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc121 params.resource_mgr = captured_state->resource_mgr(); in GetNext()
209 params.resource_mgr = new_state->resource_mgr(); in Restore()
244 params.resource_mgr = new_state->resource_mgr(); in SetIteratorFromDataset()
638 ResourceMgr resource_mgr; in DoCompute() local
639 params.resource_mgr = &resource_mgr; in DoCompute()
710 ResourceMgr resource_mgr; in DoCompute() local
711 params.resource_mgr = &resource_mgr; in DoCompute()
Dmulti_device_iterator_ops.cc108 params.resource_mgr = &resource_mgr_; in GetNextFromShard()
145 ResourceMgr* resource_mgr() { return &resource_mgr_; } in resource_mgr() function in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator
566 params.resource_mgr = resource->resource_mgr(); in Compute()
Diterator_ops.h106 ResourceMgr* resource_mgr() { return &resource_mgr_; } in resource_mgr() function
Dcaptured_function.cc1005 ResourceMgr* resource_mgr = lib_->device()->resource_manager(); in RunAsync() local
1007 f_opts.step_id, [resource_mgr](const string& name) { in RunAsync()
1008 resource_mgr->Cleanup(name).IgnoreError(); in RunAsync()
Ddataset_test_base.cc354 params.resource_mgr = op_context->resource_manager(); in CreateIteratorContext()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc238 ResourceMgr* resource_mgr = lib->device()->resource_manager(); in RunFunction() local
240 run_opts.step_id, [resource_mgr](const string& name) { in RunFunction()
241 resource_mgr->Cleanup(name).IgnoreError(); in RunFunction()
Dlookup_util.cc412 params.resource_mgr = &resource_mgr_; in Init()
/external/tensorflow/tensorflow/core/data/
Dstandalone.cc117 params.resource_mgr = &resource_mgr_; in MakeIterator()
/external/tensorflow/tensorflow/core/
DBUILD452 "//tensorflow/core/framework:resource_mgr.h",