Home
last modified time | relevance | path

Searched defs:concat_dimensions (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dconcatenation.h37 const int concat_dimensions = output_shape.DimensionsCount(); in Concatenation() local
88 const int concat_dimensions = output_shape.DimensionsCount(); in ConcatenationWithScaling() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator.cc1152 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate() local