nki_load(res, data, *[, ty, loc])
|
|
nki_store(res, data, *[, ty, loc])
|
|
nki_tensor_copy(res, data, *[, ty, loc])
|
|
nki_matmul(res, lhs, rhs[, accum, ty, loc])
|
|
nki_activation(result, data, opcode[, bias, ...])
|
|
nki_reciprocal(result, data, *[, ty, loc])
|
|
nki_tensorreduce(result, data, opcode, ...)
|
|
nki_tensortensor(result, operand0, operand1, ...)
|
|
nki_tensorscalar(result, operand0, operand1, ...)
|
|
nki_memset(result, value, *[, ty, loc])
|
|
nki_activation_reduce(reduce_res, act_res, ...)
|
|
nki_tensorscalar_reduce(reduce_res, ...[, ...])
|
|
nki_identity(result, size, *[, ty, loc])
|
|
nki_scalar_tensor_tensor(result, data, ...)
|
|
nki_scalar_tensor_scalar(result, data, ...)
|
|
nki_affine_select(result, pred, true_value, ...)
|
|