Searched refs:dims_cnt (Results 1 – 2 of 2) sorted by relevance
23 const int dims_cnt = input_shape.DimensionsCount(); in IsTranspose2DApplicable() local25 if (dims_cnt == 2) { in IsTranspose2DApplicable()32 for (int i = 1; i < dims_cnt; ++i) { in IsTranspose2DApplicable()35 rebased += dims_cnt; in IsTranspose2DApplicable()43 for (int i = 0; i < dims_cnt; ++i) { in IsTranspose2DApplicable()56 const int dims_cnt = input_shape->DimensionsCount(); in RemoveOneSizeDimensions() local57 TFLITE_DCHECK_EQ(params->perm_count, dims_cnt); in RemoveOneSizeDimensions()60 for (int i = 0; i < dims_cnt; ++i) { in RemoveOneSizeDimensions()83 for (int i = 0; i < dims_cnt; ++i) { in RemoveOneSizeDimensions()95 for (int i = 0; i < dims_cnt; ++i) { in RemoveOneSizeDimensions()
7827 const int dims_cnt = input_shape.DimensionsCount(); in TransposeImpl() local7847 if (dims_cnt == 3) { in TransposeImpl()