Home
last modified time | relevance | path

Searched refs:orig_input_dim_on_axis (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc205 int orig_input_dim_on_axis; in convertPackOp() local
209 orig_input_dim_on_axis = 1; in convertPackOp()
214 orig_input_dim_on_axis = input0_tensor_shape[concat_axis]; in convertPackOp()
217 concat_output_shape[concat_axis] = orig_input_dim_on_axis * 2; in convertPackOp()
226 concat_output_shape[concat_axis] = orig_input_dim_on_axis * (i + 1); in convertPackOp()