Searched refs:concat_type (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 218 RankedTensorType concat_type = RankedTensorType::get( in convertPackOp() local
221 op->getLoc(), concat_type, inputs[0], inputs[1], concat_axis_attr); in convertPackOp()
227 concat_type = RankedTensorType::get(ArrayRef<int64_t>(concat_output_shape), in convertPackOp()
229 a1_concat_op = rewriter.create<tosa::ConcatOp>(op->getLoc(), concat_type, in convertPackOp()
634 RankedTensorType concat_type = RankedTensorType::get( in convertConcatV2Op() local
688 op->getLoc(), concat_type, lhs_val, rhs_val, in convertConcatV2Op()
703 concat_type = RankedTensorType::get(ArrayRef<int64_t>(concat_result_shape), in convertConcatV2Op()
723 op->getLoc(), concat_type, concat_op.getResult(), rhs_val, in convertConcatV2Op()
|