Searched defs:choice_int (Results 1 – 1 of 1) sorted by relevance
457 …auto choice_int = _fused_sdp_choice_stub(at::kHIP, query_, key, value, attn_mask_, dropout_p, is_c… in _fused_sdp_choice_meta() local463 auto choice_int = _fused_sdp_choice_stub( in _fused_sdp_choice_meta() local603 int64_t choice_int = static_cast<int64_t>(sdp::SDPBackend::math); in handle_private_use() local658 int64_t choice_int = static_cast<int64_t>(sdp::SDPBackend::math); in scaled_dot_product_attention() local