Searched refs:VarDimArray (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | dim_comparator.h | 47 typedef typename gtl::ArraySlice<int64> VarDimArray; typedef
49 DimComparator(const TTypes<int64>::Matrix& ix, const VarDimArray& order, in DimComparator()
50 const VarDimArray& shape) in DimComparator()
89 const VarDimArray order_;
97 FixedDimComparator(const TTypes<int64>::Matrix& ix, const VarDimArray& order, in FixedDimComparator()
98 const VarDimArray& shape) in FixedDimComparator()
|
| D | sparse_tensor.h | 43 typedef typename gtl::ArraySlice<int64> VarDimArray; typedef
46 static Status Create(Tensor ix, Tensor vals, const VarDimArray shape,
47 const VarDimArray order, SparseTensor* result);
52 static Status Create(Tensor ix, Tensor vals, const VarDimArray shape,
56 const VarDimArray order, SparseTensor* result);
66 SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape) in SparseTensor()
72 const VarDimArray order) in SparseTensor()
77 SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape,
78 const VarDimArray order);
117 VarDimArray shape() const { return shape_; } in shape()
[all …]
|
| D | sparse_tensor.cc | 42 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()
|
| D | group_iterator.h | 79 typedef gtl::ArraySlice<int64> VarDimArray; typedef
81 GroupIterable(Tensor ix, Tensor vals, int dims, const VarDimArray& group_dims) in GroupIterable()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | set_kernels.cc | 43 using VarDimArray = sparse::SparseTensor::VarDimArray; typedef
53 Status GroupShape(const VarDimArray& input_shape, ShapeArray* grouped_shape) { in GroupShape()
86 const VarDimArray& sparse_tensor_shape) { in CheckGroup()
119 const ShapeArray Strides(const VarDimArray& shape) { in Strides()
204 const VarDimArray& input_strides, in PopulateFromDenseGroup()
227 const VarDimArray& sparse_tensor_shape, in PopulateFromSparseGroup()
272 VarDimArray group_ix = set_st.order().subspan(0, set_st.order().size() - 1); in Compute()
380 Status CheckShapesMatch(VarDimArray shape1, VarDimArray shape2) { in CheckShapesMatch()
391 Status GroupShapeFromInputs(VarDimArray shape1, VarDimArray shape2, in GroupShapeFromInputs()
403 void PopulateGroupIndices(const int64 flat_group_index, VarDimArray group_shape, in PopulateGroupIndices()
|