Searched refs:concat_dimensions (Results 1 – 2 of 2) sorted by relevance
35 const int concat_dimensions = output_shape.DimensionsCount(); in Concatenation() local36 TFLITE_DCHECK_LT(axis, concat_dimensions); in Concatenation()40 TFLITE_DCHECK_EQ(input_shapes[i]->DimensionsCount(), concat_dimensions); in Concatenation()41 for (int j = 0; j < concat_dimensions; j++) { in Concatenation()56 for (int i = axis + 1; i < concat_dimensions; ++i) { in Concatenation()86 const int concat_dimensions = output_shape.DimensionsCount(); in ConcatenationWithScaling() local87 TFLITE_DCHECK_LT(axis, concat_dimensions); in ConcatenationWithScaling()91 TFLITE_DCHECK_EQ(input_shapes[i]->DimensionsCount(), concat_dimensions); in ConcatenationWithScaling()92 for (int j = 0; j < concat_dimensions; j++) { in ConcatenationWithScaling()107 for (int i = axis + 1; i < concat_dimensions; ++i) { in ConcatenationWithScaling()
498 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate() local506 concat_dimensions[concat_dim] += in HandleConcatenate()511 reference_shape.element_type(), concat_dimensions); in HandleConcatenate()513 DimensionVector dest_indices(concat_dimensions.size(), 0); in HandleConcatenate()