Searched defs:grad_output_shape (Results 1 – 2 of 2) sorted by relevance
95 Shape grad_output_shape = ShapeUtil::MakeShape(F32, {2, 2, 2, 2}); in TEST_F() local
1912 auto grad_output_shape = gO.sizes().slice(2); in _convolution_double_backward() local