Updated on 2022-03-13 GMT+08:00

te.lang.cce.vrec(raw_tensor)

Performs the reciprocal operation 1/x on each element in a tensor. The supported types are float16 and float32. The data types int8, uint8, and int32 are converted to float16.

This API is defined in elewise_compute.py.

Parameter Description

raw_tensor: input tensor, tvm.tensor type

Return Value

res_tensor: 1/ raw_tensor, tvm.tensor type

Calling Example

shape = (1024,1024)
input_dtype = "float16"
data = tvm.placeholder(shape, name="data", dtype=input_dtype)
res = te.lang.cce.vrec(data)