Searched refs:shapes_ (Results 1 – 9 of 9) sorted by relevance
49 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()52 shapes_.emplace_back(); in Dataset()69 return shapes_; in output_shapes()91 shapes_(params.dataset->output_shapes().size()) {} in Iterator()112 shapes_[i]); in GetNextInternal()139 shapes_[i] = tensors_[i].shape(); in GetNextInternal()140 shapes_[i].RemoveDim(0); in GetNextInternal()203 shapes_[i] = tensors_[i].shape(); in RestoreInternal()204 shapes_[i].RemoveDim(0); in RestoreInternal()216 std::vector<TensorShape> shapes_ GUARDED_BY(mu_);[all …]
76 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()88 return shapes_; in output_shapes()140 TensorShape(dataset()->shapes_[i].dim_sizes())); in GetNextInternal()179 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::__anonc96147a30111::TensorSliceDatasetOp::Dataset
53 shapes_.emplace_back(t.shape().dim_sizes()); in Dataset()65 return shapes_; in output_shapes()144 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::__anon67eb3d340111::TensorDatasetOp::Dataset
38 shapes_({{-1, sparse_tensor.dims() - 1},50 return shapes_; in output_shapes()214 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::__anon2da6f3700111::Dataset
44 shapes_(shapes) { in IgniteDataset()69 return shapes_; in output_shapes()
58 const std::vector<PartialTensorShape> shapes_; variable
32 OP_REQUIRES_OK(context, context->GetAttr("shapes", &shapes_)); in ScopedAllocatorOp()41 ScopedAllocatorMgr::PopulateFields(id_, shapes_, dtype_, &fields_); in ScopedAllocatorOp()75 std::vector<TensorShape> shapes_; member in tensorflow::ScopedAllocatorOp
789 shapes_.clear(); in SnapshotNodes()816 shapes_.push_back(node->shape()); in SnapshotNodes()857 const std::vector<std::vector<int64>>& shapes() const { return shapes_; } in shapes()902 std::vector<std::vector<int64>> shapes_; variable
1720 TF_RETURN_IF_ERROR(shapes_.Merge(s1, s2)); in Merge()1740 ShapeHandle actual_shape = shapes_.GetMergedValue(shape); in AsTensorProperties()1754 DisjointSet<shape_inference::ShapeHandle> shapes_; member in tensorflow::grappler::SymbolicShapeManager