Home
last modified time | relevance | path

Searched refs:component_shapes (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtyped_queue.h36 const std::vector<TensorShape>& component_shapes,
50 const std::vector<TensorShape>& component_shapes, const string& name) in TypedQueue() argument
51 : QueueBase(capacity, component_dtypes, component_shapes, name) {} in TypedQueue()
Dqueue_base.h48 const std::vector<TensorShape>& component_shapes,
63 const std::vector<TensorShape>& component_shapes() const { in component_shapes() function
Dpadding_fifo_queue.h38 const std::vector<PartialTensorShape>& component_shapes,
Dfifo_queue.h37 const std::vector<TensorShape>& component_shapes,
Dpriority_queue.h52 const std::vector<TensorShape>& component_shapes,
Dqueue_base.cc53 const std::vector<TensorShape>& component_shapes, in QueueBase() argument
57 component_shapes_(component_shapes), in QueueBase()
Drandom_shuffle_queue_op.cc46 const std::vector<TensorShape>& component_shapes,
93 const std::vector<TensorShape>& component_shapes, const string& name) in RandomShuffleQueue() argument
94 : TypedQueue(capacity, component_dtypes, component_shapes, name), in RandomShuffleQueue()
Dbarrier_ops.cc292 const gtl::ArraySlice<TensorShape> component_shapes() const { in component_shapes() function in tensorflow::barrier::Barrier
484 if (barrier->component_shapes() != value_component_shapes_) { in VerifyResource()
487 TensorShapeUtils::ShapeListString(barrier->component_shapes()), in VerifyResource()
Dfifo_queue.cc37 const std::vector<TensorShape>& component_shapes, in FIFOQueue() argument
39 : TypedQueue(capacity, component_dtypes, component_shapes, name) {} in FIFOQueue()
Dpriority_queue.cc38 const std::vector<TensorShape>& component_shapes, in PriorityQueue() argument
40 : TypedQueue(capacity, component_dtypes, component_shapes, name) {} in PriorityQueue()