Searched refs:gradient_out (Results 1 – 2 of 2) sorted by relevance
164 XlaResource** gradient_out) { in GetOrCreateTensorArrayGradient() argument182 *gradient_out = gradient.get(); in GetOrCreateTensorArrayGradient()
120 XlaResource** gradient_out);