MCPcopy
hub / github.com/apache/tvm / output

Method output

python/tvm/te/tensor.py:101–114  ·  view source on GitHub ↗

Get the index-th output of the operation Parameters ---------- index : int The index size. Returns ------- out : Tensor The i-th output.

(self, index)

Source from the content-addressed store, hash-verified

99 """Represent an operation that generates a tensor"""
100
101 def output(self, index):
102 """Get the index-th output of the operation
103
104 Parameters
105 ----------
106 index : int
107 The index size.
108
109 Returns
110 -------
111 out : Tensor
112 The i-th output.
113 """
114 return _ffi_api.OpGetOutput(self, index)
115
116 @property
117 def num_outputs(self):

Callers 15

get_relax_matmul_moduleFunction · 0.80
computeFunction · 0.80
scanFunction · 0.80
externFunction · 0.80
mainMethod · 0.80
mainMethod · 0.80
mainMethod · 0.80
mainMethod · 0.80
mainMethod · 0.80
forwardMethod · 0.80

Calls

no outgoing calls

Tested by 15

mainMethod · 0.64
mainMethod · 0.64
mainMethod · 0.64
mainMethod · 0.64
mainMethod · 0.64
forwardMethod · 0.64
mainMethod · 0.64
mainMethod · 0.64
mainMethod · 0.64
mainMethod · 0.64
mainMethod · 0.64
mainMethod · 0.64