Home
last modified time | relevance | path

Searched refs:total_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc299 const int total_dims = sparsity->traversal_order()->size(); in VerifyAndCountSparseElements() local
302 if (total_dims < original_rank || sparsity_dim_metadata_size != total_dims) { in VerifyAndCountSparseElements()
306 const int block_rank = total_dims - original_rank; in VerifyAndCountSparseElements()
321 std::vector<int> traversal_order(total_dims); in VerifyAndCountSparseElements()
322 for (int i = 0; i < total_dims; i++) { in VerifyAndCountSparseElements()
334 for (int i = original_rank; i < total_dims; i++) { in VerifyAndCountSparseElements()
346 expanded_dim_sizes.resize(total_dims); in VerifyAndCountSparseElements()
359 if (original_block_dim < 0 || original_block_dim >= total_dims) { in VerifyAndCountSparseElements()
371 if (mapped_block_dim < 0 || mapped_block_dim >= total_dims) { in VerifyAndCountSparseElements()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc426 int total_dims = 0; in SetOpAttrList() local
441 total_dims += size; in SetOpAttrList()
446 std::unique_ptr<int64_t[]> buffer(new int64_t[total_dims]); in SetOpAttrList()
591 int total_dims = 0; in SetOpAttrListDefault() local
594 total_dims += attr.default_value().list().shape(i).dim_size(); in SetOpAttrListDefault()
598 std::unique_ptr<int64_t[]> buffer(new int64_t[total_dims]); in SetOpAttrListDefault()