Searched refs:row_shape_ (Results 1 – 1 of 1) sorted by relevance
85 row_shape_(row_shape), in Dataset()91 output_shape.AppendShape(row_shape_); in Dataset()136 row_shape_.dims()); // not an unknown rank PartialTensorShape in AsGraphDefInternal()137 for (int i = 0; i < row_shape_.dims(); i++) in AsGraphDefInternal()138 row_shape.emplace_back(row_shape_.dim_size(i)); in AsGraphDefInternal()166 DatasetIterator<Dataset<T>>::dataset()->row_shape_; in GetNextInternal()309 const PartialTensorShape row_shape_; member in tensorflow::data::__anona59be2ac0111::DenseToSparseBatchDatasetOp::Dataset