Searched defs:seed_t (Results 1 – 9 of 9) sorted by relevance
40 std::array<uint32_t, 3> CastSeedFrom(const Tensor& seed_t, const int row) { in CastSeedFrom()47 Status GetSeed(const Tensor& seed_t, const int row, in GetSeed()72 const Tensor& seed_t = context->input(1); in Compute() local
182 const Tensor& seed_t = ctx->input(0); in Compute() local211 const Tensor& seed_t = ctx->input(0); in Compute() local
203 const Tensor& seed_t = context->input(1); in Compute() local
288 const Tensor& seed_t = ctx->input(2); in Compute() local
77 const Tensor& seed_t = context->input(1); in Compute() local
22 struct seed_t struct24 seed_t(unsigned int x) in seed_t() function
464 const Tensor& seed_t = context->input(3); in Compute() local
491 at::Tensor seed_t, offset_t; in mha_fwd() local745 at::Tensor seed_t, offset_t; in mha_varlen_fwd() local
1030 at::Tensor seed_t, offset_t; in _efficient_attention_forward() local