| /external/tensorflow/tensorflow/core/kernels/lookup_tables/ |
| D | op_kernel_templates.h | 64 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), in Compute()
92 cinfo_.resource_manager()->template LookupOrCreate<ResourceType, true>( in Compute()
93 cinfo_.container(), cinfo_.name(), &resource, creator)); in Compute()
100 ctx, cinfo_.container(), cinfo_.name()); in Compute()
113 if (table_handle_set_ && cinfo_.resource_is_private_to_kernel()) { in ~ResourceConstructionOp()
114 if (!cinfo_.resource_manager() in ~ResourceConstructionOp()
115 ->template Delete<ResourceType>(cinfo_.container(), in ~ResourceConstructionOp()
116 cinfo_.name()) in ~ResourceConstructionOp()
139 return cinfo_.resource_manager()->template LookupOrCreate<T, true>( in RegisterAlias()
140 cinfo_.container(), cinfo_.name(), &alias_resource, creator); in RegisterAlias()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_op_kernel.h | 63 if (cinfo_.resource_is_private_to_kernel()) { in ~ResourceOpKernel()
64 if (!cinfo_.resource_manager() in ~ResourceOpKernel()
65 ->template Delete<T>(cinfo_.container(), cinfo_.name()) in ~ResourceOpKernel()
77 OP_REQUIRES_OK(context, cinfo_.Init(mgr, def())); in Compute()
82 mgr->LookupOrCreate<T>(cinfo_.container(), cinfo_.name(), &resource, in Compute()
100 h(0) = cinfo_.container(); in Compute()
101 h(1) = cinfo_.name(); in Compute()
107 context, 0, cinfo_.container(), cinfo_.name(), in Compute()
117 ContainerInfo cinfo_ GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/test_kernels/ |
| D | bigtable_test_client_op.cc | 29 if (cinfo_.resource_is_private_to_kernel()) { in ~BigtableTestClientOp()
30 if (!cinfo_.resource_manager() in ~BigtableTestClientOp()
31 ->Delete<BigtableClientResource>(cinfo_.container(), in ~BigtableTestClientOp()
32 cinfo_.name()) in ~BigtableTestClientOp()
42 OP_REQUIRES_OK(ctx, cinfo_.Init(mgr, def())); in Compute()
47 cinfo_.container(), cinfo_.name(), &resource, in Compute()
64 ctx, 0, cinfo_.container(), cinfo_.name(), in Compute()
70 ContainerInfo cinfo_ GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
| D | bigtable_kernels.cc | 56 if (cinfo_.resource_is_private_to_kernel()) { in ~BigtableClientOp()
57 if (!cinfo_.resource_manager() in ~BigtableClientOp()
58 ->Delete<BigtableClientResource>(cinfo_.container(), in ~BigtableClientOp()
59 cinfo_.name()) in ~BigtableClientOp()
70 OP_REQUIRES_OK(ctx, cinfo_.Init(mgr, def())); in Compute()
75 cinfo_.container(), cinfo_.name(), &resource, in Compute()
100 ctx, 0, cinfo_.container(), cinfo_.name(), in Compute()
111 ContainerInfo cinfo_ GUARDED_BY(mu_);
127 if (cinfo_.resource_is_private_to_kernel()) { in ~BigtableTableOp()
128 if (!cinfo_.resource_manager() in ~BigtableTableOp()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.h | 56 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), in Compute()
76 cinfo_.resource_manager() in Compute()
78 cinfo_.container(), cinfo_.name(), &table, creator)); in Compute()
83 DataTypeToEnum<value_dtype>::v(), cinfo_.name())); in Compute()
89 MakeResourceHandle<lookup::LookupInterface>(ctx, cinfo_.container(), in Compute()
90 cinfo_.name()); in Compute()
94 h(0) = cinfo_.container(); in Compute()
95 h(1) = cinfo_.name(); in Compute()
104 if (table_handle_set_ && cinfo_.resource_is_private_to_kernel()) { in ~LookupTableOp()
105 if (!cinfo_.resource_manager() in ~LookupTableOp()
[all …]
|
| D | conditional_accumulator_base_op.h | 69 if (accumulator_handle_set_ && cinfo_.resource_is_private_to_kernel()) { in ~ConditionalAccumulatorBaseOp()
70 TF_CHECK_OK((cinfo_.resource_manager() in ~ConditionalAccumulatorBaseOp()
72 cinfo_.container(), cinfo_.name()))); in ~ConditionalAccumulatorBaseOp()
85 ContainerInfo cinfo_; variable
91 TF_RETURN_IF_ERROR(cinfo_.Init(ctx->resource_manager(), def())); in SetAccumulatorHandle()
100 (cinfo_.resource_manager() in SetAccumulatorHandle()
102 cinfo_.container(), cinfo_.name(), &accumulator, creator))); in SetAccumulatorHandle()
109 h(0) = cinfo_.container(); in SetAccumulatorHandle()
110 h(1) = cinfo_.name(); in SetAccumulatorHandle()
|
| D | variable_ops.cc | 58 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), in Compute()
68 OP_REQUIRES_OK(ctx, cinfo_.resource_manager()->LookupOrCreate<LegacyVar>( in Compute()
69 cinfo_.container(), cinfo_.name(), &var, creator)); in Compute()
|
| D | sparse_tensors_map_ops.cc | 138 TF_RETURN_IF_ERROR(cinfo_.Init(ctx->resource_manager(), def(), in GetMap()
142 SparseTensorsMap* map = new SparseTensorsMap(cinfo_.name()); in GetMap()
148 cinfo_.resource_manager()->LookupOrCreate<SparseTensorsMap>( in GetMap()
149 cinfo_.container(), cinfo_.name(), &sparse_tensors_map_, in GetMap()
157 ContainerInfo cinfo_; member in tensorflow::SparseTensorAccessingOp
|
| D | variable_ops.h | 41 ContainerInfo cinfo_ GUARDED_BY(init_mu_);
|
| D | priority_queue_op.cc | 56 component_shapes_, cinfo_.name()); in CreateResource()
|
| D | padding_fifo_queue_op.cc | 59 capacity_, component_types_, component_shapes_, cinfo_.name()); in CreateResource()
|
| D | conditional_accumulator_op.cc | 37 new ConditionalAccumulator<Device, T>(dtype_, shape_, cinfo_.name(), in GetCreator()
|
| D | sparse_conditional_accumulator_op.cc | 38 dtype_, shape_, cinfo_.name(), reduction_type_); in GetCreator()
|
| D | barrier_ops.cc | 468 cinfo_.name()); in CreateResource()
479 "Shared barrier '", cinfo_.name(), "' has component types ", in VerifyResource()
486 "Shared barrier '", cinfo_.name(), "' has component shapes ", in VerifyResource()
|
| D | fifo_queue.cc | 380 component_shapes_, cinfo_.name()); in CreateResource()
|
| D | random_shuffle_queue_op.cc | 499 component_shapes_, cinfo_.name()); in CreateResource()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | threadpool_dataset_op.cc | 81 if (cinfo_.resource_is_private_to_kernel()) { in ~ThreadPoolHandleOp()
82 if (!cinfo_.resource_manager() in ~ThreadPoolHandleOp()
83 ->Delete<ThreadPoolResource>(cinfo_.container(), cinfo_.name()) in ~ThreadPoolHandleOp()
94 OP_REQUIRES_OK(ctx, cinfo_.Init(mgr, def())); in Compute()
97 cinfo_.container(), cinfo_.name(), &resource, in Compute()
110 ctx, 0, cinfo_.container(), cinfo_.name(), in Compute()
116 ContainerInfo cinfo_ GUARDED_BY(mu_);
|
| D | indexed_dataset_op.cc | 260 if (cinfo_.resource_is_private_to_kernel()) { in ~MaterializedHandleOp()
261 if (!cinfo_.resource_manager() in ~MaterializedHandleOp()
263 cinfo_.container(), cinfo_.name()) in ~MaterializedHandleOp()
277 OP_REQUIRES_OK(context, cinfo_.Init(mgr, def())); in Compute()
282 cinfo_.container(), cinfo_.name(), &resource, in Compute()
300 context, 0, cinfo_.container(), cinfo_.name(), in Compute()
319 ContainerInfo cinfo_; // Written once under mu_ then constant afterwards. member in tensorflow::data::__anon0a1add170111::MaterializedHandleOp
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 388 if (cinfo_.resource_is_private_to_kernel()) { in ~MultiDeviceIteratorHandleOp()
389 if (!cinfo_.resource_manager() in ~MultiDeviceIteratorHandleOp()
390 ->template Delete<MultiDeviceIterator>(cinfo_.container(), in ~MultiDeviceIteratorHandleOp()
391 cinfo_.name()) in ~MultiDeviceIteratorHandleOp()
400 string unique_name = cinfo_.name(); in Compute()
401 string container_name = cinfo_.container(); in Compute()
413 OP_REQUIRES_OK(context, cinfo_.Init(mgr, def())); in Compute()
430 unique_name = cinfo_.name(); in Compute()
431 container_name = cinfo_.container(); in Compute()
475 ContainerInfo cinfo_; // Written once under mu_ then constant afterwards. member in tensorflow::data::__anon2db6daca0111::MultiDeviceIteratorHandleOp
|
| D | iterator_ops.cc | 406 if (cinfo_.resource_is_private_to_kernel()) { in ~IteratorHandleOp()
407 if (!cinfo_.resource_manager() in ~IteratorHandleOp()
408 ->template Delete<IteratorResource>(cinfo_.container(), in ~IteratorHandleOp()
409 cinfo_.name()) in ~IteratorHandleOp()
437 OP_REQUIRES_OK(context, cinfo_.Init(mgr, def())); in Compute()
443 cinfo_.container(), cinfo_.name(), &resource, in Compute()
464 context, 0, cinfo_.container(), cinfo_.name(), in Compute()
778 if (!cinfo_.resource_manager() in ~OneShotIteratorOp()
779 ->Delete<IteratorResource>(cinfo_.container(), cinfo_.name()) in ~OneShotIteratorOp()
824 cinfo_ = cinfo; in Init()
[all …]
|
| D | iterator_ops.h | 70 ContainerInfo cinfo_; // Written once under mu_ then constant afterwards. variable
|