| /external/tensorflow/tensorflow/core/kernels/ |
| fifo_queue_op.cc | 45 OP_REQUIRES_OK(context, context->GetAttr("shapes", &component_shapes_));
52 component_shapes_, cinfo_.name());
56 std::vector<TensorShape> component_shapes_; member in class:tensorflow::FIFOQueueOp
|
| padding_fifo_queue_op.cc | 47 OP_REQUIRES_OK(context, context->GetAttr("shapes", &component_shapes_));
48 for (const auto& shape : component_shapes_) {
59 capacity_, component_types_, component_shapes_, cinfo_.name());
63 std::vector<PartialTensorShape> component_shapes_; member in class:tensorflow::PaddingFIFOQueueOp
|
| priority_queue_op.cc | 45 OP_REQUIRES_OK(context, context->GetAttr("shapes", &component_shapes_));
47 if (!component_shapes_.empty()) {
48 component_shapes_.insert(component_shapes_.begin(), TensorShape({}));
56 component_shapes_, cinfo_.name());
60 std::vector<TensorShape> component_shapes_; member in class:tensorflow::PriorityQueueOp
|
| queue_base.h | 67 return component_shapes_;
110 bool specified_shapes() const { return component_shapes_.size() > 0; }
117 shape.AppendShape(component_shapes_[i]);
147 const std::vector<TensorShape> component_shapes_; member in class:tensorflow::QueueBase
|
| random_shuffle_queue_op.cc | 491 OP_REQUIRES_OK(context, context->GetAttr("shapes", &component_shapes_));
499 component_shapes_, cinfo_.name());
506 std::vector<TensorShape> component_shapes_; member in class:tensorflow::RandomShuffleQueueOp
|
