Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc317 auto transposed_input_shape = transposed_input_type.getShape(); in convertUnpackOp() local
318 int64_t transposed_input_rank = transposed_input_shape.size(); in convertUnpackOp()
320 for (int i = 0; i < transposed_input_shape[0]; i++) { in convertUnpackOp()
329 size_vals.push_back(transposed_input_shape[j]); in convertUnpackOp()
330 shape_vals.push_back(transposed_input_shape[j]); in convertUnpackOp()