Home
last modified time | relevance | path

Searched refs:unpacked_linear_context (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/vulkan/ops/
DGru.cpp239 const c10::impl::GenericList unpacked_linear_context = in unpack() local
243 unpacked_linear_context.size() > 0u, in unpack()
247 unpacked_linear_context.get(LinearPackedContext::Unpacked::Weight) in unpack()
251 unpacked_linear_context.get(LinearPackedContext::Unpacked::Bias) in unpack()
DLstm.cpp281 const c10::impl::GenericList unpacked_linear_context = in unpack() local
285 unpacked_linear_context.size() > 0u, in unpack()
289 unpacked_linear_context.get(LinearPackedContext::Unpacked::Weight) in unpack()
293 unpacked_linear_context.get(LinearPackedContext::Unpacked::Bias) in unpack()