/external/tensorflow/tensorflow/core/distributed_runtime/ |
session_mgr_test.cc | 103 EXPECT_EQ(devices_1[0]->resource_manager(), devices_2[0]->resource_manager()); 104 EXPECT_NE(devices_1[0]->resource_manager(), devices_3[0]->resource_manager()); 105 EXPECT_NE(devices_1[0]->resource_manager(), devices_4[0]->resource_manager()); 106 EXPECT_NE(devices_3[0]->resource_manager(), devices_4[0]->resource_manager());
|
/external/tensorflow/tensorflow/core/common_runtime/ |
device_mgr.cc | 109 s.Update(dev->resource_manager()->Cleanup( 110 dev->resource_manager()->default_container())); 113 s.Update(dev->resource_manager()->Cleanup(c));
|
renamed_device.h | 119 ResourceMgr* resource_manager() override { 121 return Device::resource_manager(); 123 return underlying_->resource_manager();
|
device.h | 134 virtual ResourceMgr* resource_manager() { return rmgr_; } function in class:tensorflow::Device
|
/system/tpm/trunks/ |
trunksd.cc | 28 #include "trunks/resource_manager.h" 113 trunks::ResourceManager resource_manager(factory, low_level_transceiver); 116 base::Unretained(&resource_manager))); 118 &resource_manager, background_thread.task_runner());
|
/external/tensorflow/tensorflow/core/framework/ |
resource_op_kernel.h | 58 if (!cinfo_.resource_manager() 70 ResourceMgr* mgr = context->resource_manager();
|
resource_mgr_test.cc | 229 params.resource_manager = &resource_mgr; 259 params.resource_manager = &resource_mgr; 269 other_params.resource_manager = &other_resource_mgr; 288 params.resource_manager = &resource_mgr; 304 params.resource_manager = &resource_mgr;
|
resource_mgr.h | 276 // resource_manager(), the resource is in the container() and its 279 ResourceMgr* resource_manager() const { return rmgr_; } function in class:tensorflow::ContainerInfo 297 // ctx->resource_manager(). 434 return ctx->resource_manager()->Lookup(container, shared_name, resource); 464 return ctx->resource_manager()->Create(p.container(), p.name(), value); 471 return ctx->resource_manager()->Lookup(p.container(), p.name(), value); 478 return ctx->resource_manager()->LookupOrCreate(p.container(), p.name(), value, 485 return ctx->resource_manager()->Delete<T>(p.container(), p.name());
|
resource_op_kernel_test.cc | 119 params.resource_manager = &mgr_;
|
resource_mgr.cc | 38 actual_container = ctx->resource_manager()->default_container(); 292 return ctx->resource_manager()->Delete(p);
|
/external/tensorflow/tensorflow/core/kernels/ |
conditional_accumulator_base_op.h | 68 TF_CHECK_OK((cinfo_.resource_manager() 88 TF_RETURN_IF_ERROR(cinfo_.Init(ctx->resource_manager(), def())); 97 (cinfo_.resource_manager()
|
variable_ops.cc | 58 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), 68 OP_REQUIRES_OK(ctx, cinfo_.resource_manager()->LookupOrCreate<LegacyVar>( 97 ResourceMgr* rm = context->resource_manager(); 149 ResourceMgr* rm = context->resource_manager();
|
lookup_table_op.h | 56 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), 76 cinfo_.resource_manager() 106 cinfo_.resource_manager()->template Delete<lookup::LookupInterface>(
|
ops_testutil.h | 142 ResourceMgr* rm = device_->resource_manager(); 179 params_.get()->resource_manager = device_.get()->resource_manager();
|
lookup_util.cc | 277 return ctx->resource_manager()->Lookup(container, table_handle, table); 302 TF_RETURN_IF_ERROR(ctx->resource_manager()->Lookup(container, table_handle,
|
critical_section.cc | 186 ctx_->resource_manager()->Cleanup(name).IgnoreError();
|
sparse_tensors_map_ops.cc | 137 TF_RETURN_IF_ERROR(cinfo_.Init(ctx->resource_manager(), def(), 147 cinfo_.resource_manager()->LookupOrCreate<SparseTensorsMap>(
|
stack_ops.cc | 149 ResourceMgr* rm = ctx->resource_manager(); 194 ResourceMgr* rm = ctx->resource_manager();
|
stage_op.cc | 187 auto rm = ctx->resource_manager();
|
/external/tensorflow/tensorflow/core/kernels/data/ |
captured_function.cc | 199 ctx->lib()->device()->resource_manager()->Cleanup(name).IgnoreError(); 233 ctx->lib()->device()->resource_manager()->Cleanup(name).IgnoreError(); 277 ResourceMgr* resource_mgr = ctx->lib()->device()->resource_manager();
|
iterator_ops.cc | 452 if (!cinfo_.resource_manager() 481 ResourceMgr* mgr = context->resource_manager(); 676 if (!cinfo_.resource_manager() 736 TF_RETURN_IF_ERROR(cinfo->Init(ctx->resource_manager(), def())); 745 ctx->resource_manager()->LookupOrCreate<IteratorResource>( 777 ctx->resource_manager()->Cleanup(name).IgnoreError(); [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/ |
graph_compiler.cc | 243 params->resource_manager = device_->resource_manager();
|
xla_context.cc | 48 TF_CHECK_OK(ctx->resource_manager()->Lookup(
|
/external/tensorflow/tensorflow/contrib/data/kernels/ |
prefetching_kernels.cc | 271 OP_REQUIRES_OK(ctx, cinfo.Init(ctx->resource_manager(), def())); 275 ctx, ctx->resource_manager()->LookupOrCreate<FunctionBufferingResource>(
|
/external/tensorflow/tensorflow/c/eager/ |
runtime.cc | 282 params.resource_manager = device_->resource_manager();
|