Searched refs:aoti_torch_get_device_index (Results 1 – 3 of 3) sorted by relevance
51 aoti_torch_get_device_index(handle, &device_index)); in convert_handle_to_arrayref_tensor()
239 aoti_torch_get_device_index(AtenTensorHandle tensor, int32_t* ret_device_index);
334 AOTITorchError aoti_torch_get_device_index( in aoti_torch_get_device_index() function