Searched refs:binary_pointwise_inplace_batch_rule (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesBinaryOps.cpp | 96 void binary_pointwise_inplace_batch_rule( in binary_pointwise_inplace_batch_rule() function
469 …VMAP_SUPPORT2(sub_, Tensor, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorScalarInplaceT, &… in TORCH_LIBRARY_IMPL()
471 …VMAP_SUPPORT2(mul_, Tensor, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorInplaceT, &Tensor… in TORCH_LIBRARY_IMPL()
473 …VMAP_SUPPORT2(div_, Tensor, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorInplaceT, &Tensor… in TORCH_LIBRARY_IMPL()
474 …VMAP_SUPPORT2(div_, Tensor_mode, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorInplaceModeT… in TORCH_LIBRARY_IMPL()
476 …VMAP_SUPPORT2(clamp_min_, Tensor, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorInplaceT, &… in TORCH_LIBRARY_IMPL()
477 …VMAP_SUPPORT2(clamp_max_, Tensor, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorInplaceT, &… in TORCH_LIBRARY_IMPL()
478 …VMAP_SUPPORT2(masked_fill_, Scalar, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorScalarInp… in TORCH_LIBRARY_IMPL()
479 VMAP_SUPPORT(copy_, SINGLE_ARG(binary_pointwise_inplace_batch_rule<CopyT, &Tensor::copy_, bool>)); in TORCH_LIBRARY_IMPL()
505 SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorInplaceT, &Tensor:: op ## _ >)); in TORCH_LIBRARY_IMPL()
|