Fix custom calls.

Author: ysiraichi

torch_xla/csrc/ops/custom_call.cpp

@@ -13,7 +13,7 @@ CustomCall::CustomCall(
     xla::Shape output_shape, bool has_side_effect,
     const std::string& backend_config, const int api_version,
     const std::unordered_map<std::string, std::string>& frontend_attributes)
-    : XlaNode(xla_custom_call, inputs, std::move(output_shape),
+    : XlaNode(xla_custom_call, inputs, output_shape,
               /*num_outputs=*/output_shape.tuple_shapes_size(),
               torch::lazy::MHash(call_target)),
       call_target_(call_target),

torch_xla/csrc/ops/gpu_custom_call.cpp

@@ -9,7 +9,7 @@ namespace torch_xla {
 GpuCustomCall::GpuCustomCall(torch::lazy::OpList inputs,
                              xla::Shape output_shape,
                              const std::string& payload)
-    : XlaNode(xla_gpu_custom_call, inputs, std::move(output_shape),
+    : XlaNode(xla_gpu_custom_call, inputs, output_shape,
               /*num_outputs=*/output_shape.tuple_shapes_size(),
               torch::lazy::MHash(payload)),
       payload_(payload) {}

torch_xla/csrc/ops/tpu_custom_call.cpp

@@ -9,7 +9,7 @@ namespace torch_xla {
 TpuCustomCall::TpuCustomCall(torch::lazy::OpList inputs,
                              xla::Shape output_shape,
                              const std::string& payload)
-    : XlaNode(xla_tpu_custom_call, inputs, std::move(output_shape),
+    : XlaNode(xla_tpu_custom_call, inputs, output_shape,
               /*num_outputs=*/output_shape.tuple_shapes_size(),
               torch::lazy::MHash(payload)),
       payload_(payload) {}