Searched refs:backprop_val_grad_flat (Results 1 – 2 of 2) sorted by relevance
85 const T *backprop_val_grad_flat = backprop_val_grad->flat<T>().data(); in Compute() local106 val_grad_flat[i] = backprop_val_grad_flat[j]; in Compute()
84 const T *backprop_val_grad_flat = backprop_val_grad->flat<T>().data(); in Compute() local92 a_or_b##_val_grad_flat[idx] = backprop_val_grad_flat[k]; \ in Compute()