d1e6e624ef891543c5bba32bb0a696d350714693,coremltools/converters/nnssa/coreml/ssa_converter.py,SSAConverter,_convert_gather_nd,#SSAConverter#Any#,917

Before Change

            input_names=input_names,
            output_name=node.name
        )
        shapes.propagate_single_layer(layer, self.tensor_shapes)

    def _convert_scatter_nd(self, node):
        assert len(node.inputs) == 3
        indices, updates, shape = self._get_input_tensors(node)

After Change

        shapes.propagate_single_layer(layer, self.tensor_shapes)

    def _convert_gather_nd(self, node):
        input_nodes, input_names, input_types = self._get_input_tensors(node)
        layer = self._get_builder().add_gather_nd(
            name=node.name,
            input_names=input_names,

Instances