Searched refs:pointer_to_optional (Results 1 – 2 of 2) sorted by relevance
66 inline std::optional<T> pointer_to_optional(T* ptr) { in pointer_to_optional() function71 inline std::optional<T> pointer_to_optional(U* ptr) { in pointer_to_optional() function76 inline std::optional<at::Tensor> pointer_to_optional(AtenTensorHandle* ptr) { in pointer_to_optional() function82 inline std::optional<at::Tensor> pointer_to_optional( in pointer_to_optional() function89 inline std::optional<at::Generator> pointer_to_optional( in pointer_to_optional() function139 result.emplace_back(pointer_to_optional(ptr[i]));163 result.emplace_back(pointer_to_optional<at::Tensor>(ptr[i]));
485 pointer_to_optional( in aoti_torch__embedding_bag()534 auto optional_scale = pointer_to_optional(scale); in aoti_torch__scaled_dot_product_flash_attention_v2()619 pointer_to_optional(tensor_handle_to_tensor_pointer(attn_bias)); in aoti_torch__scaled_dot_product_efficient_attention()620 auto optional_scale = pointer_to_optional(scale); in aoti_torch__scaled_dot_product_efficient_attention()657 auto optional_bias = pointer_to_optional(bias_tensor); in aoti_torch_convolution()707 pointer_to_optional(bias_tensor), in aoti_torch__scaled_mm()708 pointer_to_optional(scale_result_tensor), in aoti_torch__scaled_mm()709 pointer_to_optional<c10::ScalarType>(out_dtype), in aoti_torch__scaled_mm()738 pointer_to_optional(bias_tensor), in aoti_torch__scaled_mm_v2()739 pointer_to_optional(scale_result_tensor), in aoti_torch__scaled_mm_v2()[all …]