Home
last modified time | relevance | path

Searched refs:component_shapes_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpriority_queue_op.cc45 OP_REQUIRES_OK(context, context->GetAttr("shapes", &component_shapes_)); in PriorityQueueOp()
47 if (!component_shapes_.empty()) { in PriorityQueueOp()
48 component_shapes_.insert(component_shapes_.begin(), TensorShape({})); in PriorityQueueOp()
56 component_shapes_, cinfo_.name()); in CreateResource()
60 std::vector<TensorShape> component_shapes_; member in tensorflow::PriorityQueueOp
Dpadding_fifo_queue_op.cc47 OP_REQUIRES_OK(context, context->GetAttr("shapes", &component_shapes_)); in PaddingFIFOQueueOp()
48 for (const auto& shape : component_shapes_) { in PaddingFIFOQueueOp()
59 capacity_, component_types_, component_shapes_, cinfo_.name()); in CreateResource()
63 std::vector<PartialTensorShape> component_shapes_; member in tensorflow::PaddingFIFOQueueOp
Dqueue_base.h64 return component_shapes_; in component_shapes()
110 bool specified_shapes() const { return component_shapes_.size() > 0; } in specified_shapes()
117 shape.AppendShape(component_shapes_[i]); in ManyOutShape()
147 const std::vector<TensorShape> component_shapes_; variable
Dtyped_queue.h58 if (!component_shapes_.empty() && in Initialize()
59 component_dtypes_.size() != component_shapes_.size()) { in Initialize()
63 ", Shapes: ", ShapeListString(component_shapes_)); in Initialize()
Dqueue_base.cc57 component_shapes_(component_shapes), in QueueBase()
132 if (requested_shapes != component_shapes_) { in MatchesNodeDefShapes()
135 ShapeListString(component_shapes_), in MatchesNodeDefShapes()
148 if (!component_shapes_[i].IsSameSize(tuple[i].shape())) { in ValidateTuple()
151 component_shapes_[i].DebugString(), ", got ", in ValidateTuple()
Dpriority_queue.cc53 if (specified_shapes() && !TensorShapeUtils::IsScalar(component_shapes_[0])) { in Initialize()
57 component_shapes_[0].DebugString()); in Initialize()
422 if (requested_shapes != component_shapes_) { in MatchesPriorityNodeDefShapes()
425 ShapeListString(component_shapes_), in MatchesPriorityNodeDefShapes()
Dfifo_queue.h85 std::vector<TensorShape> component_shapes_; variable
Drandom_shuffle_queue_op.cc492 OP_REQUIRES_OK(context, context->GetAttr("shapes", &component_shapes_)); in RandomShuffleQueueOp()
500 component_shapes_, cinfo_.name()); in CreateResource()
507 std::vector<TensorShape> component_shapes_; member in tensorflow::RandomShuffleQueueOp
Dfifo_queue.cc375 OP_REQUIRES_OK(context, context->GetAttr("shapes", &component_shapes_)); in FIFOQueueOp()
380 component_shapes_, cinfo_.name()); in CreateResource()