Searched refs:graph_outputs (Results 1 – 8 of 8) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/onnx/mx2onnx/ |
| H A D | _export_onnx.py | 224 … graph_outputs = {n: {'shape': s, 'dtype': d} for n, s, d in zip(out_names, out_shapes, out_types)}
226 return in_shapes, graph_outputs
297 in_shapes, graph_outputs = MXNetGraph.get_outputs(sym, params, in_shapes, output_label,
366 if nodename in graph_outputs:
410 elem_type=graph_outputs[nodename]['dtype'],
411 shape=graph_outputs[nodename]['shape']
420 if i.name in graph_outputs:
424 elem_type=graph_outputs[i.name]['dtype'],
425 shape=graph_outputs[i.name]['shape']
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/onnx/mx2onnx/ |
| H A D | _export_onnx.py | 224 … graph_outputs = {n: {'shape': s, 'dtype': d} for n, s, d in zip(out_names, out_shapes, out_types)}
226 return in_shapes, graph_outputs
297 in_shapes, graph_outputs = MXNetGraph.get_outputs(sym, params, in_shapes, output_label,
366 if nodename in graph_outputs:
410 elem_type=graph_outputs[nodename]['dtype'],
411 shape=graph_outputs[nodename]['shape']
420 if i.name in graph_outputs:
424 elem_type=graph_outputs[i.name]['dtype'],
425 shape=graph_outputs[i.name]['shape']
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/thirdparty/onnx/onnx/common/ |
| H A D | ir_pb_converter.cc | 505 std::unordered_set<Value*> graph_outputs(g->outputs().begin(), g->outputs().end()); in encodeGraph() local
525 if (graph_outputs.find(output) != graph_outputs.end()) { in encodeGraph()
|
| /dports/misc/py-onnx/onnx-1.10.2/onnx/common/ |
| H A D | ir_pb_converter.cc | 564 std::unordered_set<Value*> graph_outputs(g->outputs().begin(), g->outputs().end()); in encodeGraph() local
585 if (graph_outputs.find(output) != graph_outputs.end()) { in encodeGraph()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/tensorrt/ |
| H A D | nnvm_to_onnx.cc | 735 const std::vector<nnvm::IndexedGraph::NodeEntry>& graph_outputs = in GetOutputLookup() local
737 for (uint32_t i = 0; i < graph_outputs.size(); ++i) { in GetOutputLookup()
738 const uint32_t id = graph_outputs[i].node_id; in GetOutputLookup()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/tensorrt/ |
| H A D | nnvm_to_onnx.cc | 735 const std::vector<nnvm::IndexedGraph::NodeEntry>& graph_outputs = in GetOutputLookup() local
737 for (uint32_t i = 0; i < graph_outputs.size(); ++i) { in GetOutputLookup()
738 const uint32_t id = graph_outputs[i].node_id; in GetOutputLookup()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/thirdparty/onnx/onnx/test/ |
| H A D | optimizer_test.py | 41 graph_outputs = [helper.make_tensor_value_info(
44 graph_outputs.append(
47 graph_outputs)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/frontend/onnx/ |
| H A D | test_forward.py | 2957 graph_outputs = [
2966 graph_outputs.append(
2987 graph = helper.make_graph([rnn_node], "rnn_test", inputs=input_tensors, outputs=graph_outputs)
|