Searched refs:padded_shapes_ (Results 1 – 2 of 2) sorted by relevance
60 padded_shapes_(std::move(padded_shapes)), in Dataset()83 PartialTensorShape({batch_size_}).Concatenate(padded_shapes_[i]));86 PartialTensorShape({-1}).Concatenate(padded_shapes_[i]));143 padded_shapes.reserve(padded_shapes_.size()); in AsGraphDefInternal()144 for (int i = 0; i < padded_shapes_.size(); i++) { in AsGraphDefInternal()146 Tensor t(DT_INT64, TensorShape({padded_shapes_[i].dims()})); in AsGraphDefInternal()147 for (int j = 0; j < padded_shapes_[i].dims(); j++) { in AsGraphDefInternal()148 t.vec<int64_t>()(j) = padded_shapes_[i].dim_size(j); in AsGraphDefInternal()172 b->BuildAttrValue<int64_t>(padded_shapes_.size(), &N); in AsGraphDefInternal()290 dataset()->padded_shapes_[component_index]; in CopyBatch()[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::__anone4c89dd70111::PaddedBatchDatasetParams