Home
last modified time | relevance | path

Searched refs:resource_manager (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/lookup_tables/
Dop_kernel_templates.h64 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), in Compute()
92 cinfo_.resource_manager()->template LookupOrCreate<ResourceType, true>( in Compute()
114 if (!cinfo_.resource_manager() in ~ResourceConstructionOp()
139 return cinfo_.resource_manager()->template LookupOrCreate<T, true>( in RegisterAlias()
145 return cinfo_.resource_manager()->template Delete<T>(cinfo_.container(), in DeleteAlias()
188 ctx, ctx->resource_manager()->Lookup<FallbackTableBaseType, true>( in Compute()
194 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), in Compute()
227 cinfo_.resource_manager()->template LookupOrCreate<ResourceType, true>( in Compute()
249 if (!cinfo_.resource_manager() in ~TableWithFallbackConstructionOp()
276 return cinfo_.resource_manager()->template LookupOrCreate<T, true>( in RegisterAlias()
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr_test.cc136 EXPECT_EQ(devices_1[0]->resource_manager(), devices_2[0]->resource_manager()); in TEST_F()
137 EXPECT_NE(devices_1[0]->resource_manager(), devices_3[0]->resource_manager()); in TEST_F()
138 EXPECT_NE(devices_1[0]->resource_manager(), devices_4[0]->resource_manager()); in TEST_F()
139 EXPECT_NE(devices_3[0]->resource_manager(), devices_4[0]->resource_manager()); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h125 ResourceMgr* resource_manager() override { in resource_manager() function
127 return Device::resource_manager(); in resource_manager()
129 return underlying_->resource_manager(); in resource_manager()
Ddevice_mgr.cc116 s.Update(dev->resource_manager()->Cleanup( in ClearContainers()
117 dev->resource_manager()->default_container())); in ClearContainers()
120 s.Update(dev->resource_manager()->Cleanup(c)); in ClearContainers()
/external/tensorflow/tensorflow/core/kernels/
Dvariable_ops.cc58 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), in Compute()
68 OP_REQUIRES_OK(ctx, cinfo_.resource_manager()->LookupOrCreate<LegacyVar>( in Compute()
94 ResourceMgr* rm = context->resource_manager(); in Compute()
146 ResourceMgr* rm = context->resource_manager(); in Compute()
Dconditional_accumulator_base_op.h70 TF_CHECK_OK((cinfo_.resource_manager() in ~ConditionalAccumulatorBaseOp()
91 TF_RETURN_IF_ERROR(cinfo_.Init(ctx->resource_manager(), def())); in SetAccumulatorHandle()
100 (cinfo_.resource_manager() in SetAccumulatorHandle()
Dlookup_table_op.h56 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), in Compute()
76 cinfo_.resource_manager() in Compute()
105 if (!cinfo_.resource_manager() in ~LookupTableOp()
Dops_testutil.h142 ResourceMgr* rm = device_->resource_manager(); in AddResourceInput()
178 params_.get()->resource_manager = device_.get()->resource_manager(); in RunOpKernel()
Dstack.cc140 ResourceMgr* rm = ctx->resource_manager(); in GetStack()
184 ResourceMgr* rm = ctx->resource_manager(); in Compute()
/external/tensorflow/tensorflow/core/framework/
Dresource_mgr_test.cc255 params.resource_manager = &resource_mgr; in TEST()
285 params.resource_manager = &resource_mgr; in TEST()
295 other_params.resource_manager = &other_resource_mgr; in TEST()
314 params.resource_manager = &resource_mgr; in TEST()
330 params.resource_manager = &resource_mgr; in TEST()
Dresource_op_kernel.h64 if (!cinfo_.resource_manager() in ~ResourceOpKernel()
76 ResourceMgr* mgr = context->resource_manager(); in Compute()
Dresource_mgr.h328 ResourceMgr* resource_manager() const { return rmgr_; } in resource_manager() function
548 return ctx->resource_manager()->Lookup(container, shared_name, resource);
578 return ctx->resource_manager()->Create(p.container(), p.name(), value);
585 return ctx->resource_manager()->Lookup<T, use_dynamic_cast>(p.container(),
599 return ctx->resource_manager()->LookupMany(containers_and_names, values);
606 return ctx->resource_manager()->LookupOrCreate(p.container(), p.name(), value,
613 return ctx->resource_manager()->Delete<T>(p.container(), p.name());
Dresource_mgr.cc44 actual_container = ctx->resource_manager()->default_container(); in MakeResourceHandle()
305 return ctx->resource_manager()->Delete(p); in DeleteResource()
/external/tensorflow/tensorflow/contrib/bigtable/kernels/test_kernels/
Dbigtable_test_client_op.cc30 if (!cinfo_.resource_manager() in ~BigtableTestClientOp()
41 ResourceMgr* mgr = ctx->resource_manager(); in Compute()
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_kernels.cc57 if (!cinfo_.resource_manager() in ~BigtableClientOp()
69 ResourceMgr* mgr = ctx->resource_manager(); in Compute()
128 if (!cinfo_.resource_manager() in ~BigtableTableOp()
140 ResourceMgr* mgr = ctx->resource_manager(); in Compute()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc156 device_->resource_manager()->Cleanup(name).IgnoreError(); in Run()
169 device->resource_manager()->Cleanup(name).IgnoreError(); in Run()
236 params.resource_manager = device_->resource_manager(); in Run()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc307 lib_->device()->resource_manager()->Cleanup(name).IgnoreError(); in Run()
343 lib_->device()->resource_manager()->Cleanup(name).IgnoreError(); in RunWithBorrowedArgs()
379 lib_->device()->resource_manager()->Cleanup(name).IgnoreError(); in RunInstantiated()
420 ResourceMgr* resource_mgr = lib_->device()->resource_manager(); in RunAsync()
Diterator_ops.cc407 if (!cinfo_.resource_manager() in ~IteratorHandleOp()
436 ResourceMgr* mgr = context->resource_manager(); in Compute()
516 ResourceMgr* mgr = context->resource_manager(); in Compute()
778 if (!cinfo_.resource_manager() in ~OneShotIteratorOp()
838 TF_RETURN_IF_ERROR(cinfo->Init(ctx->resource_manager(), def())); in TryInit()
847 ctx->resource_manager()->LookupOrCreate<IteratorResource>( in TryInit()
880 ctx->resource_manager()->Cleanup(name).IgnoreError(); in TryInit()
Dsingle_threaded_executor.cc218 params.resource_manager = device->resource_manager(); in RunAsync()
Ddataset_test_base.cc183 params_->resource_manager = device_->resource_manager(); in CreateOpKernelContext()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_device.cc29 *rm = ctx->resource_manager(); in GetResourceManager()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dget_serialized_resource_op.cc47 OP_REQUIRES_OK(context, context->resource_manager()->Lookup( in Compute()
Dget_serialized_resource_op_test.cc54 ResourceMgr* rm = device->resource_manager(); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc310 params->resource_manager = device_->resource_manager(); in PartiallySetupParams()
/external/drm_hwcomposer/
Ddrmdisplaycompositor.h49 int Init(ResourceManager *resource_manager, int display);

123