Searched refs:reshaped_result_dims (Results 1 – 1 of 1) sorted by relevance
278 std::vector<int64> reshaped_result_dims(warp_dims.begin(), in CalculateGradData() local280 reshaped_result_dims.push_back(2); in CalculateGradData()281 reshaped_result_dims.push_back(2); in CalculateGradData()285 reshaped_result_dims, broadcasted_dims, in CalculateGradData()