Home
last modified time | relevance | path

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

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