For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo struct
NanCheckCountOp
struct NanCheckCountOp
Counts NaN/Inf values in a floating-point tensor.
See nn.nan_check for implementation details.
Implemented traits
Methods
execute
static def execute[dtype: DType, rank: Int, target: StringSlice[ImmStaticOrigin], _trace_name: StringSlice[ImmStaticOrigin] = StringSlice("")](nan_count_out: ManagedTensorSlice[IOSpec[_, _].Output, static_spec=nan_count_out.static_spec], inf_count_out: ManagedTensorSlice[IOSpec[_, _].Output, static_spec=inf_count_out.static_spec], input: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec], ctx: DeviceContext)