Home
last modified time | relevance | path

Searched refs:enforce_safe_casting_to_output (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.cpp886 .enforce_safe_casting_to_output(true) \
969 .enforce_safe_casting_to_output(true) in build_ternary_op()
984 .enforce_safe_casting_to_output(true) \
1008 .enforce_safe_casting_to_output(true) \
1029 .enforce_safe_casting_to_output(false) \
H A DTensorIterator.h870 TensorIteratorConfig& enforce_safe_casting_to_output( in enforce_safe_casting_to_output() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DActivation.cpp96 .enforce_safe_casting_to_output(true)); in TORCH_META_FUNC()
110 .enforce_safe_casting_to_output(true)); in TORCH_META_FUNC()
H A DLoss.cpp450 .enforce_safe_casting_to_output(true) in smooth_l1_loss_backward_out()
H A DTensorCompare.cpp137 .enforce_safe_casting_to_output(true) in TORCH_META_FUNC2()
H A DTensorShape.cpp625 .enforce_safe_casting_to_output(true) in TORCH_IMPL_FUNC()
654 .enforce_safe_casting_to_output(true) in TORCH_IMPL_FUNC()
H A DLinearAlgebra.cpp1189 .enforce_safe_casting_to_output(true) in build_addr_iter()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator.cu336 .enforce_safe_casting_to_output(true) in CompileAndLaunchKernel()