Searched refs:out_grad_shape (Results 1 – 1 of 1) sorted by relevance
2409 DimVector out_grad_shape( in matchAndRewrite() local2429 /*output_size=*/out_grad_shape[dim], in matchAndRewrite()2450 out_grad_shape[dim] = low_padding[dim] + high_padding[dim] + in matchAndRewrite()2451 (out_grad_shape[dim] - 1) * strides[dim] + 1; in matchAndRewrite()2454 loc, RankedTensorType::get(out_grad_shape, element_type), in matchAndRewrite()