Searched defs:lazy_output (Results 1 – 1 of 1) sorted by relevance
1186 auto lazy_output = torch::sort(lazy_input, /*dim=*/0, /*descending=*/true); in TEST_F() local1243 torch::Tensor lazy_output = torch::min(lazy_input); in TEST_F() local1254 torch::Tensor lazy_output = torch::max(lazy_input); in TEST_F() local2050 torch::Tensor lazy_output = torch::group_norm( in TEST_F() local2142 torch::Tensor lazy_output = torch::instance_norm( in TEST_F() local2184 torch::Tensor lazy_output = torch::layer_norm( in TEST_F() local2265 torch::Tensor lazy_output = in TEST_F() local2285 torch::Tensor lazy_output = in TEST_F() local2308 torch::Tensor lazy_output = torch::cosine_embedding_loss( in TEST_F() local2329 torch::Tensor lazy_output = torch::hinge_embedding_loss( in TEST_F() local[all …]