Searched defs:concat_dimensions (Results 1 – 2 of 2) sorted by relevance
461 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate() local
1609 const int concat_dimensions = output_shape.DimensionsCount(); in Concatenation() local1662 const int concat_dimensions = output_shape.DimensionsCount(); in ConcatenationWithScaling() local2231 const int concat_dimensions = input_shape.DimensionsCount(); in Split() local