Home
last modified time | relevance | path

Searched refs:component_dtypes (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dqueue_op.cc96 for (DataType dt : queue->component_dtypes()) { in ComputeAsync()
137 for (DataType dt : queue->component_dtypes()) { in ComputeAsync()
170 ctx, ctx->MatchSignature({DT_RESOURCE}, queue->component_dtypes()), in ComputeAsync()
174 ctx, ctx->MatchSignature({DT_STRING_REF}, queue->component_dtypes()), in ComputeAsync()
222 ctx->MatchSignature({DT_RESOURCE, DT_INT32}, queue->component_dtypes()), in ComputeAsync()
227 queue->component_dtypes()), in ComputeAsync()
295 ctx->MatchSignature({DT_RESOURCE, DT_INT32}, queue->component_dtypes()), in ComputeAsync()
300 queue->component_dtypes()), in ComputeAsync()
Dtyped_queue.h35 TypedQueue(const int32 capacity, const DataTypeVector& component_dtypes,
49 int32 capacity, const DataTypeVector& component_dtypes, in TypedQueue() argument
51 : QueueBase(capacity, component_dtypes, component_shapes, name) {} in TypedQueue()
Dqueue_base.h51 QueueBase(int32 capacity, const DataTypeVector& component_dtypes,
56 const DataTypeVector& component_dtypes() const override { in component_dtypes() function
Dpadding_fifo_queue.h37 PaddingFIFOQueue(int32 capacity, const DataTypeVector& component_dtypes,
Dfifo_queue.h36 FIFOQueue(int32 capacity, const DataTypeVector& component_dtypes,
Dpriority_queue.h51 PriorityQueue(int32 capacity, const DataTypeVector& component_dtypes,
Drandom_shuffle_queue_op.cc45 int64 seed2, const DataTypeVector& component_dtypes,
91 const DataTypeVector& component_dtypes, in RandomShuffleQueue() argument
93 : TypedQueue(capacity, component_dtypes, component_shapes, name), in RandomShuffleQueue()
Dqueue_base.cc52 QueueBase::QueueBase(int32 capacity, const DataTypeVector& component_dtypes, in QueueBase() argument
56 component_dtypes_(component_dtypes), in QueueBase()
Dfifo_queue.cc36 FIFOQueue::FIFOQueue(int capacity, const DataTypeVector& component_dtypes, in FIFOQueue() argument
39 : TypedQueue(capacity, component_dtypes, component_shapes, name) {} in FIFOQueue()
Dpriority_queue.cc37 const DataTypeVector& component_dtypes, in PriorityQueue() argument
40 : TypedQueue(capacity, component_dtypes, component_shapes, name) {} in PriorityQueue()
Dpadding_fifo_queue.cc37 int capacity, const DataTypeVector& component_dtypes, in PaddingFIFOQueue() argument
39 : FIFOQueue(capacity, component_dtypes, in PaddingFIFOQueue()
/external/tensorflow/tensorflow/contrib/checkpoint/python/
Dsplit_dependency.py94 def split_dependency(component_names, component_dtypes, argument
127 for name, dtype in zip(component_names, component_dtypes):
Dsplit_dependency_test.py53 component_dtypes=(self.combined.dtype,) * 2,
/external/tensorflow/tensorflow/core/framework/
Dqueue_interface.h90 virtual const DataTypeVector& component_dtypes() const = 0;
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py769 component_dtypes=(dtype,) * len(self._param_names),