Searched defs:grad_values (Results 1 – 4 of 4) sorted by relevance
652 ShapeHandle grad_values = c->input(1); in __anonc55240e61902() local
402 const auto grad_values = grad_values_t->vec<T>(); in Compute() local
490 GpuLaunchConfig cfg, const Tindex* reverse_index_map, const T* grad_values, in GatherOriginalGradValuesKernel()
1474 grad_values, argument