Home
last modified time | relevance | path

Searched refs:VarDimArray (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/
H A Dsparse_tensor.h44 typedef typename gtl::ArraySlice<int64_t> VarDimArray; typedef
47 static Status Create(Tensor ix, Tensor vals, const VarDimArray shape,
48 const VarDimArray order, SparseTensor* result);
53 static Status Create(Tensor ix, Tensor vals, const VarDimArray shape,
57 const VarDimArray order, SparseTensor* result);
67 SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape) in SparseTensor()
73 const VarDimArray order) in SparseTensor()
78 SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape,
79 const VarDimArray order);
118 VarDimArray shape() const { return shape_; } in shape()
[all …]
H A Ddim_comparator.h47 typedef typename gtl::ArraySlice<int64_t> VarDimArray; typedef
49 DimComparator(const TTypes<int64_t>::Matrix& ix, const VarDimArray& order, in DimComparator()
50 const VarDimArray& shape) in DimComparator()
90 const VarDimArray order_;
99 const VarDimArray& order, const VarDimArray& shape) in FixedDimComparator()
H A Dsparse_tensor.cc42 const VarDimArray shape, in Create()
43 const VarDimArray order, in Create()
84 const VarDimArray shape, in Create()
92 const VarDimArray order, in Create()
98 SparseTensor::SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape, in SparseTensor()
99 const VarDimArray order) in SparseTensor()
H A Dgroup_iterator.h79 typedef gtl::ArraySlice<int64_t> VarDimArray; typedef
81 GroupIterable(Tensor ix, Tensor vals, int dims, const VarDimArray& group_dims) in GroupIterable()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dset_kernels.cc45 using VarDimArray = sparse::SparseTensor::VarDimArray; typedef
55 Status GroupShape(const VarDimArray& input_shape, ShapeArray* grouped_shape) { in GroupShape()
96 const VarDimArray& sparse_tensor_shape) { in CheckGroup()
129 const ShapeArray Strides(const VarDimArray& shape) { in Strides()
216 const VarDimArray& input_strides, in PopulateFromDenseGroup()
239 const VarDimArray& sparse_tensor_shape, in PopulateFromSparseGroup()
283 VarDimArray group_ix = set_st.order().subspan(0, set_st.order().size() - 1); in Compute()
419 Status CheckShapesMatch(VarDimArray shape1, VarDimArray shape2) { in CheckShapesMatch()
430 Status GroupShapeFromInputs(VarDimArray shape1, VarDimArray shape2, in GroupShapeFromInputs()
443 VarDimArray group_shape, in PopulateGroupIndices()