| /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 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_ TF_GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.h | 63 OP_REQUIRES_OK(ctx, cinfo_.Init(ctx->resource_manager(), def(), in Compute()
85 cinfo_.resource_manager() in Compute()
87 cinfo_.container(), cinfo_.name(), &table, creator)); in Compute()
92 DataTypeToEnum<value_dtype>::v(), cinfo_.name())); in Compute()
99 ctx, cinfo_.container(), cinfo_.name()); in Compute()
105 h(0) = cinfo_.container(); in Compute()
106 h(1) = cinfo_.name(); in Compute()
115 if (table_handle_set_ && cinfo_.resource_is_private_to_kernel()) { in ~LookupTableOp()
116 if (!cinfo_.resource_manager() in ~LookupTableOp()
117 ->template Delete<lookup::LookupInterface>(cinfo_.container(), 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()
91 ContainerInfo cinfo_; variable
100 TF_RETURN_IF_ERROR(cinfo_.Init(ctx->resource_manager(), def())); in SetAccumulatorHandle()
108 (cinfo_.resource_manager() in SetAccumulatorHandle()
110 cinfo_.container(), cinfo_.name(), &accumulator, creator))); in SetAccumulatorHandle()
117 h(0) = cinfo_.container(); in SetAccumulatorHandle()
118 h(1) = cinfo_.name(); in SetAccumulatorHandle()
|
| D | conditional_accumulator_op.cc | 37 new ConditionalAccumulator<Device, T>(dtype_, shape_, cinfo_.name(), in GetCreator()
74 new ConditionalAccumulator<Device, T>(dtype_, shape_, cinfo_.name(), in GetCreator()
89 h(0) = cinfo_.container(); in SetHandleToOutput()
90 h(1) = cinfo_.name(); in SetHandleToOutput()
92 ctx, 0, cinfo_.container(), cinfo_.name(), in SetHandleToOutput()
|
| D | variable_ops.cc | 71 OP_REQUIRES_OK(context, cinfo_.Init(context->resource_manager(), def(), in VariableOp()
82 OP_REQUIRES_OK(ctx, cinfo_.resource_manager()->LookupOrCreate<LegacyVar>( in Compute()
83 cinfo_.container(), cinfo_.name(), &var, creator)); in Compute()
|
| D | sparse_tensors_map_ops.cc | 139 TF_RETURN_IF_ERROR(cinfo_.Init(ctx->resource_manager(), def(), in GetMap()
143 SparseTensorsMap* map = new SparseTensorsMap(cinfo_.name()); in GetMap()
149 cinfo_.resource_manager()->LookupOrCreate<SparseTensorsMap>( in GetMap()
150 cinfo_.container(), cinfo_.name(), &sparse_tensors_map_, in GetMap()
158 ContainerInfo cinfo_; member in tensorflow::SparseTensorAccessingOp
|
| D | variable_ops.h | 39 ContainerInfo cinfo_; variable
|
| 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 | 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 | 500 component_shapes_, cinfo_.name()); in CreateResource()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | threadpool_dataset_op.cc | 84 if (cinfo_.resource_is_private_to_kernel()) { in ~ThreadPoolHandleOp()
85 if (!cinfo_.resource_manager() in ~ThreadPoolHandleOp()
86 ->Delete<ThreadPoolResource>(cinfo_.container(), cinfo_.name()) in ~ThreadPoolHandleOp()
97 OP_REQUIRES_OK(ctx, cinfo_.Init(mgr, def())); in Compute()
100 cinfo_.container(), cinfo_.name(), &resource, in Compute()
113 ctx, 0, cinfo_.container(), cinfo_.name(), in Compute()
119 ContainerInfo cinfo_ TF_GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 458 if (cinfo_.resource_is_private_to_kernel()) { in ~IteratorHandleOp()
459 if (!cinfo_.resource_manager() in ~IteratorHandleOp()
460 ->template Delete<IteratorResource>(cinfo_.container(), in ~IteratorHandleOp()
461 cinfo_.name()) in ~IteratorHandleOp()
490 OP_REQUIRES_OK(context, cinfo_.Init(mgr, def())); in Compute()
496 cinfo_.container(), cinfo_.name(), &resource, in Compute()
517 context, 0, cinfo_.container(), cinfo_.name(), in Compute()
792 if (!cinfo_.resource_manager() in ~OneShotIteratorOp()
793 ->Delete<IteratorResource>(cinfo_.container(), cinfo_.name()) in ~OneShotIteratorOp()
840 cinfo_ = cinfo; in Init()
[all …]
|
| D | multi_device_iterator_ops.cc | 411 if (cinfo_.resource_is_private_to_kernel()) { in ~MultiDeviceIteratorHandleOp()
412 if (!cinfo_.resource_manager() in ~MultiDeviceIteratorHandleOp()
413 ->template Delete<MultiDeviceIterator>(cinfo_.container(), in ~MultiDeviceIteratorHandleOp()
414 cinfo_.name()) in ~MultiDeviceIteratorHandleOp()
423 string unique_name = cinfo_.name(); in Compute()
424 string container_name = cinfo_.container(); in Compute()
436 OP_REQUIRES_OK(context, cinfo_.Init(mgr, def())); in Compute()
453 unique_name = cinfo_.name(); in Compute()
454 container_name = cinfo_.container(); in Compute()
498 ContainerInfo cinfo_; // Written once under mu_ then constant afterwards. member in tensorflow::data::__anon2e5a99280111::MultiDeviceIteratorHandleOp
|
| D | iterator_ops.h | 165 ContainerInfo cinfo_; // Written once under mu_ then constant afterwards. variable
|