Searched defs:reverse_index_map (Results 1 – 3 of 3) sorted by relevance
134 Tindex* reverse_index_map) { in ScatterInputElementsKernel()193 Tindex* reverse_index_map; in operator ()() local377 Tindex* reverse_index_map; in operator ()() local490 GpuLaunchConfig cfg, const Tindex* reverse_index_map, const T* grad_values, in GatherOriginalGradValuesKernel()
69 Tindex* reverse_index_map = nullptr; in operator ()() local385 const auto reverse_index_map = reverse_index_map_t->vec<Tindex>(); in Compute() local
632 ShapeHandle reverse_index_map = c->Vector(N); in __anon9213d3ed1802() local647 ShapeHandle reverse_index_map = c->input(0); in __anon9213d3ed1902() local