Home
last modified time | relevance | path

Searched refs:partial_shapes_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpadding_fifo_queue.cc41 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()
Dpadding_fifo_queue.h72 std::vector<PartialTensorShape> partial_shapes_; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op.cc50 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
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc490 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