Home
last modified time | relevance | path

Searched refs:convert_tensor_to_scalar_list (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DForeachPointwiseOp.cu238 auto scalars = convert_tensor_to_scalar_list(scalars_, input.size()); \
254 auto scalars = convert_tensor_to_scalar_list(scalars_, input.size()); \
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DForeachOpsKernels.cpp315 auto scalars = convert_tensor_to_scalar_list(scalars_, input.size()); \
326 auto scalars = convert_tensor_to_scalar_list(scalars_, input.size()); \
H A DForeachUtils.h218 inline std::vector<c10::Scalar> convert_tensor_to_scalar_list( in convert_tensor_to_scalar_list() function