Searched defs:input_packed (Results 1 – 3 of 3) sorted by relevance
320 Array2D<uint32_t> input_packed = PackLargeInput<uint8_t>(input); in XLA_TEST_F() local334 Array2D<uint32_t> input_packed = PackLargeInput<uint8_t>(input); in XLA_TEST_F() local
598 auto input_packed = input_inner.packed_accessor64<const scalar_t, 5>(); in TORCH_IMPL_FUNC() local
6578 def in_proj(input_packed, qkv_linear=qkv_linear): argument