Searched defs:input_grad (Results 1 – 3 of 3) sorted by relevance
56 SparseGradient input_grad({grad.data(), indices.data(), 6}); in TEST_F() local108 SparseGradient input_grad({grad.data(), indices.data(), 6}); in TEST_F() local
56 SparseGradient<int> input_grad({grad.data(), indices.data(), 6}); in TEST_F() local108 SparseGradient<int> input_grad({grad.data(), indices.data(), 6}); in TEST_F() local
172 auto input_grad = param.input_grad_; in SplitAndCalculateSegmentBucketSize() local