Searched refs:_run_layernorm_correctness_test (Results 1 – 1 of 1) sorted by relevance
387 def _run_layernorm_correctness_test(layer, dtype='float32'): function499 _run_layernorm_correctness_test(504 _run_layernorm_correctness_test(