Method forward

imperative/python/megengine/module/quantized/concat.py:16–18 · view source on GitHub ↗

(self, inps: Iterable[Tensor], axis: int = 0)

Source from the content-addressed store, hash-verified

14	self.output_dtype = dtype
15
16	def forward(self, inps: Iterable[Tensor], axis: int = 0):
17	new_inps = tuple(x.astype(self.output_dtype) for x in inps)
18	return F.concat(new_inps, axis)
19
20	@classmethod
21	def from_qat_module(cls, qat_module: QAT.Concat):

nothing calls this directly

concatMethod · 0.80

astypeMethod · 0.45

no test coverage detected