Searched defs:mask_data (Results 1 – 9 of 9) sorted by relevance
192 bool* mask_data = nullptr; in __anonba9190ed0202() local301 bool* mask_data = nullptr; in host_softmax_backward() local
26 auto mask_data = mask_contig.data_ptr<bool>(); in _rowwise_prune_helper() local
43 std::vector<std::vector<int>> mask_data; in TEST() local
144 std::vector<int> mask_data(num_elements); in cross_attention_mask() local
62 const bool* const mask_data = mask.const_data_ptr<bool>(); in masked_select_out() local
134 std::unique_ptr<uint32_t[]> mask_data(new uint32_t[width * height]); in CreateMouseCursorFromHCursor() local
263 mask_t* mask_data = has_attn_mask in cpu_flash_attention() local526 mask_t* mask_data = has_attn_mask in cpu_flash_attention_backward() local
384 auto mask_data = mask_cont.const_data_ptr<bool>(); in launch_masked_scatter_kernel() local
400 const accum_t* mask_data = in cpu_flash_attention() local