Searched defs:philox_state (Results 1 – 3 of 3) sorted by relevance
48 xla::XlaOp philox_state = in GetBitGeneratorForDevice() local
500 at::PhiloxCudaState philox_state = gen->philox_cuda_state(counter_offset); in mha_fwd() local753 at::PhiloxCudaState philox_state = gen->philox_cuda_state(counter_offset); in mha_varlen_fwd() local
1042 at::PhiloxCudaState philox_state; in _efficient_attention_forward() local