Searched defs:z_shape (Results 1 – 2 of 2) sorted by relevance
128 TensorShape z_shape({4, 2}); in TEST() local
582 TensorShape z_shape = shapes[2]; in TestMatMulGrad() local629 TensorShape z_shape; in RandMatMulShapes() local