Lines Matching full:numel
158 ASSERT_EQ(tensor.numel(), 1); in TEST()
163 ASSERT_EQ(tensor.numel(), 1); in TEST()
168 ASSERT_EQ(tensor.numel(), 1); in TEST()
173 ASSERT_EQ(tensor.numel(), 1); in TEST()
178 ASSERT_EQ(tensor.numel(), 1); in TEST()
185 ASSERT_EQ(tensor.numel(), 1); in TEST()
190 ASSERT_EQ(tensor.numel(), 1); in TEST()
197 ASSERT_EQ(tensor.numel(), 1); in TEST()
204 ASSERT_EQ(tensor.numel(), 3); in TEST()
211 ASSERT_EQ(tensor.numel(), 3); in TEST()
218 ASSERT_EQ(tensor.numel(), 3); in TEST()
228 ASSERT_EQ(tensor.numel(), 3); in TEST()
238 ASSERT_EQ(tensor.numel(), 3); in TEST()
245 ASSERT_EQ(tensor.numel(), 3); in TEST()
253 ASSERT_EQ(tensor.numel(), 3); in TEST()
263 ASSERT_EQ(tensor.numel(), 3); in TEST()
273 ASSERT_EQ(tensor.numel(), 3); in TEST()
281 ASSERT_EQ(tensor.numel(), v.size()); in TEST()
289 ASSERT_EQ(tensor.numel(), w.size()); in TEST()
307 ASSERT_EQ(tensor.numel(), x.size()); in TEST()
317 ASSERT_EQ(tensor.numel(), 3); in TEST()
324 ASSERT_EQ(tensor.numel(), 3); in TEST()
331 ASSERT_EQ(tensor.numel(), 1); in TEST()
358 ASSERT_EQ(tensor.numel(), 1); in TEST()
369 ASSERT_EQ(tensor.numel(), 1); in test_TorchTensorCtorScalarFloatingType_expected_dtype()
375 ASSERT_EQ(tensor.numel(), 1); in test_TorchTensorCtorScalarFloatingType_expected_dtype()
381 ASSERT_EQ(tensor.numel(), 1); in test_TorchTensorCtorScalarFloatingType_expected_dtype()
396 ASSERT_EQ(tensor.numel(), 1); in TEST()
402 ASSERT_EQ(tensor.numel(), 1); in TEST()
410 ASSERT_EQ(tensor.numel(), 3); in TEST()
418 ASSERT_EQ(tensor.numel(), 3); in TEST()
426 ASSERT_EQ(tensor.numel(), 3); in TEST()
434 ASSERT_EQ(tensor.numel(), 3); in TEST()
442 ASSERT_EQ(tensor.numel(), 3); in TEST()
455 ASSERT_EQ(tensor.numel(), 3); in test_TorchTensorCtorSingleDimFloatingType_expected_dtype()
463 ASSERT_EQ(tensor.numel(), 3); in test_TorchTensorCtorSingleDimFloatingType_expected_dtype()
471 ASSERT_EQ(tensor.numel(), 3); in test_TorchTensorCtorSingleDimFloatingType_expected_dtype()
478 ASSERT_EQ(tensor.numel(), 3); in test_TorchTensorCtorSingleDimFloatingType_expected_dtype()
486 ASSERT_EQ(tensor.numel(), 3); in test_TorchTensorCtorSingleDimFloatingType_expected_dtype()
493 ASSERT_EQ(tensor.numel(), 3); in test_TorchTensorCtorSingleDimFloatingType_expected_dtype()
510 ASSERT_EQ(tensor.numel(), 3); in TEST()
518 ASSERT_EQ(tensor.numel(), 3); in TEST()
692 ASSERT_EQ(tensor.numel(), 3); in TEST()
699 ASSERT_EQ(tensor.numel(), 3); in TEST()
706 ASSERT_EQ(tensor.numel(), 1); in TEST()
766 ASSERT_EQ(tensor.numel(), 0); in test_TorchTensorCtorZeroSizedDim_expected_dtype()
773 ASSERT_EQ(tensor.numel(), 0); in test_TorchTensorCtorZeroSizedDim_expected_dtype()
780 ASSERT_EQ(tensor.numel(), 0); in test_TorchTensorCtorZeroSizedDim_expected_dtype()
787 ASSERT_EQ(tensor.numel(), 0); in test_TorchTensorCtorZeroSizedDim_expected_dtype()
794 ASSERT_EQ(tensor.numel(), 0); in test_TorchTensorCtorZeroSizedDim_expected_dtype()
801 ASSERT_EQ(tensor.numel(), 0); in test_TorchTensorCtorZeroSizedDim_expected_dtype()
808 ASSERT_EQ(tensor.numel(), 0); in test_TorchTensorCtorZeroSizedDim_expected_dtype()
997 ASSERT_EQ(tensor.numel(), 3); in TEST()
1032 ASSERT_EQ(tensor.numel(), 9); in TEST()
1187 ASSERT_EQ(x.var(0).numel(), 3); in TEST()
1188 ASSERT_EQ(x.std(0).numel(), 3); in TEST()
1190 ASSERT_EQ(x.var(0, /*unbiased=*/true).numel(), 3); in TEST()
1191 ASSERT_EQ(x.std(0, /*unbiased=*/true).numel(), 3); in TEST()
1193 ASSERT_EQ(torch::var(x, 0).numel(), 3); in TEST()
1194 ASSERT_EQ(std::get<0>(torch::var_mean(x, 0)).numel(), 3); in TEST()
1195 ASSERT_EQ(torch::std(x, 0).numel(), 3); in TEST()
1196 ASSERT_EQ(std::get<0>(torch::std_mean(x, 0)).numel(), 3); in TEST()
1198 ASSERT_EQ(torch::var(x, 0, /*unbiased=*/true).numel(), 3); in TEST()
1199 ASSERT_EQ(std::get<0>(torch::var_mean(x, 0, /*unbiased=*/true)).numel(), 3); in TEST()
1200 ASSERT_EQ(torch::std(x, 0, /*unbiased=*/true).numel(), 3); in TEST()
1201 ASSERT_EQ(std::get<0>(torch::std_mean(x, 0, /*unbiased=*/true)).numel(), 3); in TEST()