Home
last modified time | relevance | path

Searched refs:computeUpdatedLinearWeightAndBias (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfold_linear_bn.h24 TORCH_API std::tuple<at::Tensor, at::Tensor> computeUpdatedLinearWeightAndBias(
H A Dfold_linear_bn.cpp13 std::tuple<at::Tensor, at::Tensor> computeUpdatedLinearWeightAndBias( in computeUpdatedLinearWeightAndBias() function
H A Dfrozen_linear_folding.cpp114 computeUpdatedLinearWeightAndBias(params); in FoldFrozenLinearBatchnorm()