From cd94bbf672387a9080631ca869115ee4f51da300 Mon Sep 17 00:00:00 2001 From: Ranjit Ranjan Date: Thu, 4 May 2023 09:46:46 -0500 Subject: [PATCH 6/7] AIX support for GatherToSplitFusion::ApplyImpl --- onnxruntime/core/optimizer/gather_fusion.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/onnxruntime/core/optimizer/gather_fusion.cc b/onnxruntime/core/optimizer/gather_fusion.cc index ad6dd3c20..7ae2ff18d 100644 --- a/onnxruntime/core/optimizer/gather_fusion.cc +++ b/onnxruntime/core/optimizer/gather_fusion.cc @@ -174,7 +174,11 @@ Status GatherToSplitFusion::ApplyImpl(Graph& graph, bool& modified, int graph_le axes_initializer_proto.set_data_type(ONNX_NAMESPACE::TensorProto_DataType_INT64); InlinedVector axes_value{split_axis}; axes_initializer_proto.set_raw_data(axes_value.data(), axes_value.size() * sizeof(int64_t)); + if constexpr (endian::native != endian::little) { + utils::ConvertRawDataInTensorProto((ONNX_NAMESPACE::TensorProto*)&axes_initializer_proto); + } NodeArg* axes_arg = &graph_utils::AddInitializer(graph, axes_initializer_proto); + for (size_t i = 0; i < output_count; ++i) { Node& squeeze_node = -- 2.38.3