Searched refs:grad_output_indices (Results 1 – 1 of 1) sorted by relevance
292 std::vector<int64> grad_output_indices(warp_dims_without_last_dims.size()); in CalculateGradData() local293 std::iota(grad_output_indices.begin(), grad_output_indices.end(), 0); in CalculateGradData()294 grad_output_indices.push_back(weights_with_channels_dims.size() - 1); in CalculateGradData()296 grad_output, weights_with_channels_dims, grad_output_indices); in CalculateGradData()