| /aosp_15_r20/external/tensorflow/tensorflow/lite/testdata/ |
| H A D | sparse_tensor.json | 24 "dense_size": 2 number
35 "dense_size": 2 number
39 "dense_size": 2 number
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_to_dense_op_gpu.cu.cc | 55 __global__ void SetDefaultValue(const T default_value, const int64 dense_size, in SetDefaultValue()
111 Status LaunchComputeKernels(OpKernelContext* c, const int64 dense_size, in LaunchComputeKernels()
153 const int64 dense_size = dense->NumElements(); in operator ()() local
|
| H A D | reshape_util.cc | 107 const int64_t dense_size = input_shape.num_elements(); in ReshapeSparseTensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/utils/ |
| H A D | sparsity_format_converter.cc | 221 const std::vector<int>& dense_size, in FormatConverter()
236 std::vector<int> dense_size(sparsity.dim_metadata_size); in FormatConverter() local
256 std::vector<TfLiteDimensionType> format, std::vector<int> dense_size, in InitSparseToDenseConverter()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseTensorImpl.h | 216 auto dense_size = size.slice(sparse_dim); in _resize_() local
270 auto dense_size = sizes().slice(sparse_dim); in resize_and_clear_() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/ |
| H A D | SparseCsrLinearAlgebra.cpp | 192 IntArrayRef dense_size) { in sparse_mm_mkl_template()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| H A D | parse_example_test.cc | 193 const char* text_def, int dense_size = 2) { in ParseExampleOpModel()
|
| H A D | parse_example.cc | 656 int dense_size = 0; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorTransformerFunctions.cpp | 41 auto dense_size = dense_matrix.size(dim_constraint); in check_nested_tensor_matrix_constraints() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | xnnpack_delegate.cc | 5043 const size_t dense_size = context->tensors[t].bytes / sizeof(float); in PrepareOpsToDelegate() local
5053 const size_t dense_size = in PrepareOpsToDelegate() local
5065 const size_t dense_size = in PrepareOpsToDelegate() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | common.h | 366 int dense_size; member
|
| /aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/ |
| H A D | common.h | 379 int dense_size; member
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 1967 def test_shape(dense_size, sparse_dims_shape, dense_dims_shape, sparse_size): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate.cc | 5291 size_t dense_size; in DensifyAndDequantizeConstTensor() local
|