Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dnormalization.h127 struct TORCH_API NormalizeFuncOptions { struct
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dnormalization.h45 NormalizeFuncOptions options = {}) {
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp1545 auto norm = F::normalize(input, F::NormalizeFuncOptions().p(1).dim(-1)); in TEST_F()
1561 F::normalize(input, F::NormalizeFuncOptions().p(1).dim(-1).out(output)); in TEST_F()
1571 F::normalize(input, F::NormalizeFuncOptions().p(1).dim(-1)); in TEST_F()