Skip to content

Commit 4b76e4c

Browse files
committed
rename _get_raw_layer_by_name
1 parent 43f0abf commit 4b76e4c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

nn/base.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -642,10 +642,10 @@ def get_extern_data(data: Data) -> Tensor:
642642
if not data.batch:
643643
data.batch = scope.global_batch
644644
root_layer_name = f"data:{data.name}"
645-
return _get_special_layer(root_layer_name, scope=scope, data=data)
645+
return _get_raw_layer_by_name(root_layer_name, scope=scope, data=data)
646646

647647

648-
def _get_special_layer(name: str, *, scope: Optional[nn.NameCtx] = None, data: Data) -> Tensor:
648+
def _get_raw_layer_by_name(name: str, *, scope: Optional[nn.NameCtx] = None, data: Data) -> Tensor:
649649
"""
650650
Special layer can be "data:..." or whatever.
651651
"""

0 commit comments

Comments
 (0)