Home
last modified time | relevance | path

Searched refs:device_with_index (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dpromoted_prim_ops.h35 void device_with_index(Stack& stack);
H A Dpromoted_prim_ops.cpp97 void device_with_index(Stack& stack) { in device_with_index() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_prim_ops.cpp2424 device_with_index,