Searched defs:reverse_index_map (Results 1 – 4 of 4) 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 __anonad4a10b71802() local650 ShapeHandle reverse_index_map = c->input(0); in __anonad4a10b71902() local
2031 Py_ssize_t *reverse_index_map = NULL; in remove_unused_consts() local