Searched defs:reverse_index_map (Results 1 – 3 of 3) sorted by relevance
133 Tindex* reverse_index_map) { in ScatterInputElementsKernel()192 Tindex* reverse_index_map; in operator ()() local376 Tindex* reverse_index_map; in operator ()() local490 GpuLaunchConfig cfg, const Tindex* reverse_index_map, const T* grad_values, in GatherOriginalGradValuesKernel()
73 Tindex* reverse_index_map = nullptr; in operator ()() local401 const auto reverse_index_map = reverse_index_map_t->vec<Tindex>(); in Compute() local
635 ShapeHandle reverse_index_map = c->Vector(N); in __anonc55240e61802() local650 ShapeHandle reverse_index_map = c->input(0); in __anonc55240e61902() local