Searched defs:func_out (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | gradcheck.py | 673 fn, inputs, inp_indices, func_out, all_u, all_v, eps, is_forward_ad argument
970 func, inputs, func_out, eps, *, is_forward_ad argument
987 func, func_out, all_inputs, inputs_indices, all_u, eps, nondet_tol argument
1442 func_out, argument
1576 func_out, argument
1864 func_out, argument
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | grad_mode.cpp | 13 torch::Tensor func_out = c * c; in TEST() local
|
| H A D | inference_mode.cpp | 103 torch::Tensor func_out = functional_op(c); // go through kernels: CPU in TEST() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UnaryOps.cpp | 301 #define CREATE_UNARY_TORCH_IMPL_FUNC(func_out, func_stub) \ argument
308 #define CREATE_UNARY_TORCH_IMPL_INTEGER_NO_OP_FUNC(func_out, func_stub) … argument
|
| H A D | BinaryOps.cpp | 530 #define CREATE_BINARY_TORCH_IMPL_FUNC(func_out, func_stub) … argument
|