Searched refs:partial_shapes_ (Results 1 – 4 of 4) sorted by relevance
41 partial_shapes_(partial_shapes) {} in PaddingFIFOQueue()47 if (component_dtypes_.size() != partial_shapes_.size()) { in Initialize()50 component_dtypes_.size(), " dtypes and ", partial_shapes_.size(), in Initialize()168 .Concatenate(partial_shapes_[i]); in TryDequeueMany()241 if (!partial_shapes_[i].IsCompatibleWith(tuple[i].shape())) { in ValidateTuple()244 partial_shapes_[i].DebugString(), ", got ", in ValidateTuple()257 PartialTensorShape({batch_size}).Concatenate(partial_shapes_[i]); in ValidateManyTuple()273 partial_shapes_)) { in CompatibleNodeDefShapes()276 PartialTensorShapeUtils::PartialShapeListString(partial_shapes_), in CompatibleNodeDefShapes()
72 std::vector<PartialTensorShape> partial_shapes_; variable
50 partial_shapes_.emplace_back(element_dim_sizes); in Dataset()71 return partial_shapes_; in output_shapes()172 std::vector<PartialTensorShape> partial_shapes_; member in tensorflow::data::TensorSliceDatasetOp::Dataset
490 partial_shapes_.emplace_back(element_dim_sizes); in NestedDataset()496 return partial_shapes_; in output_shapes()528 std::vector<PartialTensorShape> partial_shapes_; member in tensorflow::data::snapshot_util::Reader::NestedDataset