Searched refs:padded_shapes_ (Results 1 – 2 of 2) sorted by relevance
59 padded_shapes_(std::move(padded_shapes)), in Dataset()81 PartialTensorShape({batch_size_}).Concatenate(padded_shapes_[i]));84 PartialTensorShape({-1}).Concatenate(padded_shapes_[i]));141 padded_shapes.reserve(padded_shapes_.size()); in AsGraphDefInternal()142 for (int i = 0; i < padded_shapes_.size(); i++) { in AsGraphDefInternal()144 Tensor t(DT_INT64, TensorShape({padded_shapes_[i].dims()})); in AsGraphDefInternal()145 for (int j = 0; j < padded_shapes_[i].dims(); j++) { in AsGraphDefInternal()146 t.vec<int64>()(j) = padded_shapes_[i].dim_size(j); in AsGraphDefInternal()170 b->BuildAttrValue<int64>(padded_shapes_.size(), &N); in AsGraphDefInternal()247 dataset()->padded_shapes_[component_index]; in GetNextInternal()[all …]
38 padded_shapes_(std::move(padded_shapes)), in PaddedBatchDatasetParams()54 for (auto& padded_shape : padded_shapes_) { in GetInputTensors()97 std::vector<Tensor> padded_shapes_; member in tensorflow::data::__anon1aa214390111::PaddedBatchDatasetParams