Searched refs:concat_dimensions (Results 1 – 2 of 2) sorted by relevance
1609 const int concat_dimensions = output_shape.DimensionsCount(); in Concatenation() local1610 TFLITE_DCHECK_LT(axis, concat_dimensions); in Concatenation()1614 TFLITE_DCHECK_EQ(input_shapes[i]->DimensionsCount(), concat_dimensions); in Concatenation()1615 for (int j = 0; j < concat_dimensions; j++) { in Concatenation()1630 for (int i = axis + 1; i < concat_dimensions; ++i) { in Concatenation()1662 const int concat_dimensions = output_shape.DimensionsCount(); in ConcatenationWithScaling() local1663 TFLITE_DCHECK_LT(axis, concat_dimensions); in ConcatenationWithScaling()1667 TFLITE_DCHECK_EQ(input_shapes[i]->DimensionsCount(), concat_dimensions); in ConcatenationWithScaling()1668 for (int j = 0; j < concat_dimensions; j++) { in ConcatenationWithScaling()1683 for (int i = axis + 1; i < concat_dimensions; ++i) { in ConcatenationWithScaling()[all …]
461 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate() local469 concat_dimensions[concat_dim] += in HandleConcatenate()474 reference_shape.element_type(), concat_dimensions); in HandleConcatenate()476 DimensionVector dest_indices(concat_dimensions.size(), 0); in HandleConcatenate()