From 0aac598778c246f12d3159a28c7c74b73ef95d5f Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 17 Sep 2019 18:49:58 +0200
Subject: [PATCH] Sort parameters in Tensorflow python bindings
---
src/pystencils_autodiff/backends/python_bindings.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/pystencils_autodiff/backends/python_bindings.py b/src/pystencils_autodiff/backends/python_bindings.py
index 140fbf0..d67f276 100644
--- a/src/pystencils_autodiff/backends/python_bindings.py
+++ b/src/pystencils_autodiff/backends/python_bindings.py
@@ -94,8 +94,8 @@ REGISTER_KERNEL_BUILDER(Name("{{ python_name }}").Device({{ device }}), {{ pytho
'"tensorflow/core/framework/op_kernel.h"']
def __init__(self, function_node: KernelFunction):
- input_fields = list(function_node.fields_read)
- output_fields = list(function_node.fields_written)
+ input_fields = list(sorted(function_node.fields_read, key=lambda x: str(x)))
+ output_fields = list(sorted(function_node.fields_written, key=lambda x: str(x)))
input_field_names = [f.name for f in input_fields]
output_field_names = [f.name for f in output_fields]
parameters = function_node.get_parameters()
--
GitLab