• Home
  • Raw
  • Download

Lines Matching full:instance_norm

1685   InstanceNorm1d instance_norm(  in TEST_F()  local
1688 ASSERT_TRUE(instance_norm->options.track_running_stats()); in TEST_F()
1690 ASSERT_TRUE(instance_norm->running_mean.defined()); in TEST_F()
1691 ASSERT_EQ(instance_norm->running_mean.dim(), 1); in TEST_F()
1692 ASSERT_EQ(instance_norm->running_mean.size(0), 5); in TEST_F()
1694 ASSERT_TRUE(instance_norm->running_var.defined()); in TEST_F()
1695 ASSERT_EQ(instance_norm->running_var.dim(), 1); in TEST_F()
1696 ASSERT_EQ(instance_norm->running_var.size(0), 5); in TEST_F()
1698 ASSERT_TRUE(instance_norm->num_batches_tracked.defined()); in TEST_F()
1699 ASSERT_EQ(instance_norm->num_batches_tracked.dim(), 0); in TEST_F()
1701 ASSERT_TRUE(instance_norm->options.affine()); in TEST_F()
1703 ASSERT_TRUE(instance_norm->weight.defined()); in TEST_F()
1704 ASSERT_EQ(instance_norm->weight.dim(), 1); in TEST_F()
1705 ASSERT_EQ(instance_norm->weight.size(0), 5); in TEST_F()
1707 ASSERT_TRUE(instance_norm->bias.defined()); in TEST_F()
1708 ASSERT_EQ(instance_norm->bias.dim(), 1); in TEST_F()
1709 ASSERT_EQ(instance_norm->bias.size(0), 5); in TEST_F()
1713 InstanceNorm1d instance_norm( in TEST_F() local
1716 ASSERT_FALSE(instance_norm->running_mean.defined()); in TEST_F()
1717 ASSERT_FALSE(instance_norm->running_var.defined()); in TEST_F()
1718 ASSERT_FALSE(instance_norm->num_batches_tracked.defined()); in TEST_F()
1719 ASSERT_FALSE(instance_norm->weight.defined()); in TEST_F()
1720 ASSERT_FALSE(instance_norm->bias.defined()); in TEST_F()
1724 InstanceNorm1d instance_norm(5); in TEST_F() local
1725 instance_norm->eval(); in TEST_F()
1728 auto output = instance_norm->forward(input); in TEST_F()
1748 InstanceNorm2d instance_norm( in TEST_F() local
1751 ASSERT_TRUE(instance_norm->options.track_running_stats()); in TEST_F()
1753 ASSERT_TRUE(instance_norm->running_mean.defined()); in TEST_F()
1754 ASSERT_EQ(instance_norm->running_mean.dim(), 1); in TEST_F()
1755 ASSERT_EQ(instance_norm->running_mean.size(0), 5); in TEST_F()
1757 ASSERT_TRUE(instance_norm->running_var.defined()); in TEST_F()
1758 ASSERT_EQ(instance_norm->running_var.dim(), 1); in TEST_F()
1759 ASSERT_EQ(instance_norm->running_var.size(0), 5); in TEST_F()
1761 ASSERT_TRUE(instance_norm->num_batches_tracked.defined()); in TEST_F()
1762 ASSERT_EQ(instance_norm->num_batches_tracked.dim(), 0); in TEST_F()
1764 ASSERT_TRUE(instance_norm->options.affine()); in TEST_F()
1766 ASSERT_TRUE(instance_norm->weight.defined()); in TEST_F()
1767 ASSERT_EQ(instance_norm->weight.dim(), 1); in TEST_F()
1768 ASSERT_EQ(instance_norm->weight.size(0), 5); in TEST_F()
1770 ASSERT_TRUE(instance_norm->bias.defined()); in TEST_F()
1771 ASSERT_EQ(instance_norm->bias.dim(), 1); in TEST_F()
1772 ASSERT_EQ(instance_norm->bias.size(0), 5); in TEST_F()
1776 InstanceNorm2d instance_norm( in TEST_F() local
1779 ASSERT_FALSE(instance_norm->running_mean.defined()); in TEST_F()
1780 ASSERT_FALSE(instance_norm->running_var.defined()); in TEST_F()
1781 ASSERT_FALSE(instance_norm->num_batches_tracked.defined()); in TEST_F()
1782 ASSERT_FALSE(instance_norm->weight.defined()); in TEST_F()
1783 ASSERT_FALSE(instance_norm->bias.defined()); in TEST_F()
1787 InstanceNorm2d instance_norm(5); in TEST_F() local
1788 instance_norm->eval(); in TEST_F()
1792 auto output = instance_norm->forward(input); in TEST_F()
1812 InstanceNorm3d instance_norm( in TEST_F() local
1815 ASSERT_TRUE(instance_norm->options.track_running_stats()); in TEST_F()
1817 ASSERT_TRUE(instance_norm->running_mean.defined()); in TEST_F()
1818 ASSERT_EQ(instance_norm->running_mean.dim(), 1); in TEST_F()
1819 ASSERT_EQ(instance_norm->running_mean.size(0), 5); in TEST_F()
1821 ASSERT_TRUE(instance_norm->running_var.defined()); in TEST_F()
1822 ASSERT_EQ(instance_norm->running_var.dim(), 1); in TEST_F()
1823 ASSERT_EQ(instance_norm->running_var.size(0), 5); in TEST_F()
1825 ASSERT_TRUE(instance_norm->num_batches_tracked.defined()); in TEST_F()
1826 ASSERT_EQ(instance_norm->num_batches_tracked.dim(), 0); in TEST_F()
1828 ASSERT_TRUE(instance_norm->options.affine()); in TEST_F()
1830 ASSERT_TRUE(instance_norm->weight.defined()); in TEST_F()
1831 ASSERT_EQ(instance_norm->weight.dim(), 1); in TEST_F()
1832 ASSERT_EQ(instance_norm->weight.size(0), 5); in TEST_F()
1834 ASSERT_TRUE(instance_norm->bias.defined()); in TEST_F()
1835 ASSERT_EQ(instance_norm->bias.dim(), 1); in TEST_F()
1836 ASSERT_EQ(instance_norm->bias.size(0), 5); in TEST_F()
1840 InstanceNorm3d instance_norm( in TEST_F() local
1843 ASSERT_FALSE(instance_norm->running_mean.defined()); in TEST_F()
1844 ASSERT_FALSE(instance_norm->running_var.defined()); in TEST_F()
1845 ASSERT_FALSE(instance_norm->num_batches_tracked.defined()); in TEST_F()
1846 ASSERT_FALSE(instance_norm->weight.defined()); in TEST_F()
1847 ASSERT_FALSE(instance_norm->bias.defined()); in TEST_F()
1851 InstanceNorm3d instance_norm(5); in TEST_F() local
1852 instance_norm->eval(); in TEST_F()
1856 auto output = instance_norm->forward(input); in TEST_F()