Searched full:scalaropt_dim (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ts_native_functions.yaml | 101 - norm.ScalarOpt_dim
|
| H A D | ReduceOps.cpp | 326 TORCH_META_FUNC2(norm, ScalarOpt_dim) in TORCH_META_FUNC2() argument
|
| H A D | native_functions.yaml | 6707 - func: norm.ScalarOpt_dim(Tensor self, Scalar? p, int[1] dim, bool keepdim=False) -> Tensor
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 446 - aten::norm.ScalarOpt_dim
917 aten::norm.ScalarOpt_dim: 4
|
| H A D | model_ops.yaml | 271 aten::norm.ScalarOpt_dim: 8
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
| H A D | tts_angular_training.txt | 40 Operator: aten.norm.ScalarOpt_dim
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesReduceOps.cpp | 490 REDUCTION_BOXED_ARGS(norm.ScalarOpt_dim, 2, KEEPDIM_CASE_VARIABLE, 3); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.h | 896 "norm.ScalarOpt_dim", \
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | ops.cpp | 2171 … "aten::norm.ScalarOpt_dim(Tensor self, Scalar? p, int[1] dim, bool keepdim=False) -> Tensor"))) { in __anon11f46a8b6802()
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | derivatives.yaml | 1275 - name: norm.ScalarOpt_dim(Tensor self, Scalar? p, int[1] dim, bool keepdim=False) -> Tensor
|