Searched refs:weights_with_channels_dims (Results 1 – 1 of 1) sorted by relevance
270 std::vector<int64> weights_with_channels_dims = reshaped_weights_dims; in CalculateGradData() local271 weights_with_channels_dims.push_back(data_channels); in CalculateGradData()290 reshaped_weights, weights_with_channels_dims, reshaped_weights_indices); 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()