Home
last modified time | relevance | path

Searched defs:linear_w (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfold_linear_bn.h8 at::Tensor linear_w; member
H A Dfrozen_linear_folding.cpp58 auto linear_w = constant_as<Tensor>(linear->namedInput("weight")).value(); in FoldFrozenLinearBatchnorm() local