Searched refs:row_shape_ (Results 1 – 1 of 1) sorted by relevance
83 row_shape_(row_shape), in Dataset()89 output_shape.AppendShape(row_shape_); in Dataset()144 row_shape_.dims()); // not an unknown rank PartialTensorShape in AsGraphDefInternal()145 for (int i = 0; i < row_shape_.dims(); i++) in AsGraphDefInternal()146 row_shape.emplace_back(row_shape_.dim_size(i)); in AsGraphDefInternal()174 DatasetIterator<Dataset<T>>::dataset()->row_shape_; in GetNextInternal()318 const PartialTensorShape row_shape_; member in tensorflow::data::experimental::__anoncb2181d10111::DenseToSparseBatchDatasetOp::Dataset