Searched refs:dense_shape_ (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix.h | 109 dense_shape_(rhs.dense_shape_), in CSRSparseMatrix()
119 dense_shape_(std::move(rhs.dense_shape_)), in CSRSparseMatrix()
134 dense_shape_ = std::move(rhs.dense_shape_);
161 return ValidateTypesAndShapes(metadata_.dtype, dense_shape_, in Validate()
168 dense_shape_ = Tensor(); in Clear()
177 return metadata_.validated && dense_shape_.IsInitialized() && in valid()
180 dense_shape_.NumElements() > 1 && in valid()
191 return dense_shape_.NumElements(); in dims()
211 return dense_shape_; in dense_shape()
216 return dense_shape_; in dense_shape()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | sparse_tensor_slice_dataset_op.cc | 93 dense_shape_(DT_INT64, {params.dataset->sparse_tensor_.dims() - 1}), in Iterator()
96 for (size_t i = 0; i < dense_shape_.NumElements(); ++i) {
97 dense_shape_.vec<int64>()(i) =
146 out_tensors->push_back(dense_shape_); in GetNextInternal()
154 out_tensors->push_back(dense_shape_); in GetNextInternal()
207 Tensor dense_shape_; member in tensorflow::data::__anon1b965c8e0111::Dataset::Iterator
|
| D | sparse_tensor_slice_dataset_op_test.cc | 35 dense_shape_(std::move(dense_shape)),
41 return {indices_, values_, dense_shape_}; in GetInputTensors()
63 Tensor dense_shape_; member in tensorflow::data::__anon116fed2d0111::SparseTensorSliceDatasetParams
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 45 dense_shape_ = AddInput(TensorType_INT32); in EmbeddingLookupSparseOpModel()
62 PopulateTensor(dense_shape_, dense_shape_data); in SetInput()
87 int dense_shape_; member in tflite::__anonc8d704f40111::EmbeddingLookupSparseOpModel
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter.cc | 57 : dense_shape_(shape), in FormatConverter()
84 int num_original_dims = dense_shape_.size(); in DenseToSparse()
99 shape_offset[i - 1] = shape_offset[i] * dense_shape_[i]; in DenseToSparse()
218 : dense_shape_(shape) { in FormatConverter()
263 int orig_rank = dense_shape_.size(); in Populate()
279 dest_data[GetFlattenedIndex(orig_idx, dense_shape_)] = in Populate()
|
| D | format_converter.h | 83 std::vector<int> dense_shape_; variable
|