Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc98 ArrayRef<int64_t> input0_tensor_shape = input_type.getShape(); in convertPackOp() local
99 int input_tensor_rank = input0_tensor_shape.size(); in convertPackOp()
114 for (int d = 0; d < input0_tensor_shape.size(); d++) { in convertPackOp()
115 if (input0_tensor_shape[d] != next_tensor_shape[d]) { in convertPackOp()
212 concat_output_shape.push_back(input0_tensor_shape[i]); in convertPackOp()
214 orig_input_dim_on_axis = input0_tensor_shape[concat_axis]; in convertPackOp()