Searched defs:z_shape (Results 1 – 3 of 3) sorted by relevance
78 TensorShape z_shape({2, 3}); in TEST_F() local88 TensorShape z_shape({2, 3}); in TEST_F() local102 TensorShape z_shape({3, 2, 3}); in TEST_F() local112 TensorShape z_shape({}); in TEST_F() local122 TensorShape z_shape({4}); in TEST_F() local132 TensorShape z_shape({}); in TEST_F() local142 TensorShape z_shape({3, 5}); in TEST_F() local152 TensorShape z_shape({3, 5}); in TEST_F() local
613 TensorShape z_shape = shapes[2]; in TestMatMulGrad() local660 TensorShape z_shape; in RandMatMulShapes() local
128 TensorShape z_shape({4, 2}); in TEST() local