Searched refs:dim_metadata_size (Results 1 – 9 of 9) sorted by relevance
454 const size_t dim_metadata_size = src_sparsity->dim_metadata()->size(); in ParseSparsity() local455 sparsity->dim_metadata_size = dim_metadata_size; in ParseSparsity()457 malloc(dim_metadata_size * sizeof(TfLiteDimensionMetadata))); in ParseSparsity()459 dim_metadata_size * sizeof(TfLiteDimensionMetadata)); in ParseSparsity()461 for (int i = 0; i < dim_metadata_size; i++) { in ParseSparsity()
439 ASSERT_EQ(t1->sparsity->dim_metadata_size, 4); in TEST()
227 format_.resize(sparsity.dim_metadata_size); in FormatConverter()228 dim_metadata_.resize(2 * sparsity.dim_metadata_size); in FormatConverter()229 for (int i = 0; i < sparsity.dim_metadata_size; i++) { in FormatConverter()
129 t.sparsity->dim_metadata_size = 2; in TEST()
126 for (; i < sparsity->dim_metadata_size; i++) { in TfLiteSparsityFree()
366 int dim_metadata_size; member
116 PyObject* dim_metadata = PyList_New(param.dim_metadata_size); in PyDictFromSparsityParam()117 for (int i = 0; i < param.dim_metadata_size; i++) { in PyDictFromSparsityParam()
869 if (sparsity.dim_metadata_size == kDimMetadataSizeRandomSparse) { in EvalFloat()877 } else if (sparsity.dim_metadata_size == kDimMetadataSizeBlockSparse && in EvalFloat()
583 ->dim_metadata[tensor->sparsity->dim_metadata_size - 1] in ExtractVector()