Home
last modified time | relevance | path

Searched defs:calculate_gain (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_unsupported_ops.py63 def calculate_gain(): function
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/
H A Dinit.cpp59 double calculate_gain(NonlinearityType nonlinearity, double param) { in calculate_gain() function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dinit.py72 def calculate_gain(nonlinearity, param=None): function