||
&&
seq_len != 0 \|\| num_heads != 0
seq_len != 0 && num_heads != 0
(dim < self.dim() \|\| dim >= 0)
(dim >= 0 && dim < self.dim())
shape[2]
dim()
shape[1]
self.size(3)
output_size[1]
size() >= 1
size() >= 2
temp_perm[perm[i]]
TORCH_CHECK(perm[i] >= 0 && perm[i] < perm_len)
self.size(self.dim()-2)
TORCH_CHECK(self.dim() >= 2 && mat2.dim() >= 2)
dim.back()
TORCH_CHECK(!dim.empty())
info.pop()
if info is None: raise RuntimeError
os.environ.get('PYTORCH_VERSION')
None
.split()
SmallVector
IntArrayRef
std::string(method).data()
std::string method_str
GetApiFuncAddr
return nullptr
int
int64_t
int64_t(1)
int m, n
TORCH_CHECK(size <= INT64_MAX / next_dim)
TORCH_CHECK(ret == 0)
c10::value_or_else
right
side
searchsorted_pre_check
out
RuntimeError
TORCH_WARN_ONCE
op_plugin/utils/
SearchsortedSideUtil.h
resolve_searchsorted_effective_right
searchsorted_pre_check_npu
SearchsortedValidateUtil.h
BucketizationUtils.h
out_int32
searchsorted_validate_tensor_out_op
searchsorted_validate_core_no_output
searchsorted_validate_scalar_op
SearchsortedWarnUtil.h
searchsorted_tensor_is_row_major_contiguous
warn_if_searchsorted_*_noncontiguous
SearchsortedKernelNpu.cpp
CheckOut
SearchSorted
right_eff
op_plugin_functions.yaml
searchsorted.Scalar
searchsorted.Tensor
searchsorted.Tensor_out
gen_opapi
new_params.right_eff
(validate, warn, resolve)
bool
Tensor
structured_inherit
exec
op_api_common_base.h
#include
new_params
op_plugin::
npu_cpp_extension.h
test/test_base_ops/test_searchsorted.py
op_plugin/utils/StdVarCorrectionUtils.h
correction_fits_aclnn_int64
correction
op_plugin/ops/aclops/StdKernelNpu.cpp
std_out
std
std_mean
op_plugin/ops/aclops/VarKernelNpu.cpp
var_out
var
var_mean
test_std.py
test_var.py
