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