Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DBinaryOp.cpp15 static Tensor binary_op_scalar( in binary_op_scalar() function
391 return binary_op_scalar( in add_scalar()
459 return binary_op_scalar( in sub_scalar()
494 return binary_op_scalar( in mul_scalar()
514 return binary_op_scalar( in div_scalar()
549 return binary_op_scalar( in pow_tensor_scalar()
562 return binary_op_scalar( in pow_scalar_tensor()
569 return binary_op_scalar( in floor_divide_scalar()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DForeachFunctors.cuh114 __device__ __forceinline__ void binary_op_scalar( in binary_op_scalar() function
225 binary_op_scalar<res_arg_index>( in operator ()()
249 binary_op_scalar<res_arg_index>( in operator ()()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DBinaryOpsKernel.cpp31 inline Vectorized<scalar_t> binary_op_scalar( in binary_op_scalar() function
141 return binary_op_scalar( in mul_kernel()
173 return binary_op_scalar( in div_true_kernel()
221 return binary_op_scalar( in div_trunc_kernel()
299 return binary_op_scalar( in div_floor_kernel()