floor

Description

Rounds down raw_tensor element-wise.

Prototype

floor(raw_tensor)

Parameters

raw_tensor: a tvm.tensor for the input tensor

Atlas training product: supports float16 and float32.

Atlas inference product: supports float16, float32, f162s8, f162u8, f162s32, and f322s32.

Atlas 200I/500 A2 inference product: supports float16.

Atlas A2 training product/Atlas A2 inference product: supports float16 and float32.

Atlas A3 training product/Atlas A3 inference product: supports float16 and float32.

Returns

res_tensor: result Tensor, tvm.tensor type

The default data type is int32. Data is output based on the specified output data type. For example, if the input data type is f162s8, the output data type is int8.

Restrictions

None

Applicability

Atlas training product

Atlas inference product

Atlas 200I/500 A2 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas A3 training product/Atlas A3 inference product

Example

from tbe import tvm
from tbe import dsl
shape = (1024,1024)
input_dtype = "float16"
data = tvm.placeholder(shape, name="data", dtype=input_dtype)
res = dsl.floor(data)