Searched refs:scatter_elements (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/py-onnx/onnx-1.10.2/onnx/version_converter/adapters/ |
| H A D | scatter_10_11.h | 20 Node* scatter_elements = graph->create(kScatterElements); in adapt_scatter_10_11() local
21 scatter_elements->i_(kaxis, axis); in adapt_scatter_10_11()
22 scatter_elements->addInput(node->inputs()[0]); in adapt_scatter_10_11()
23 scatter_elements->addInput(node->inputs()[1]); in adapt_scatter_10_11()
24 scatter_elements->addInput(node->inputs()[2]); in adapt_scatter_10_11()
25 node->replaceAllUsesWith(scatter_elements); in adapt_scatter_10_11()
27 scatter_elements->insertBefore(node); in adapt_scatter_10_11()
30 return scatter_elements; in adapt_scatter_10_11()
|
| /dports/misc/py-onnx/onnx-1.10.2/onnx/backend/test/case/node/ |
| H A D | scatterelements.py | 16 def scatter_elements(data, indices, updates, axis=0): # type: ignore function
63 y = scatter_elements(data, indices, updates)
85 y = scatter_elements(data, indices, updates, axis)
105 y = scatter_elements(data, indices, updates, axis)
|
| /dports/misc/py-onnx-tf/onnx-tf-1.6.0/onnx_tf/handlers/backend/ |
| H A D | scatter.py | 4 from onnx_tf.handlers.backend.scatter_elements import ScatterElements
|