Searched defs:layer_fn (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | models.py | 73 def _make_new_nodes(nodes_by_depth, layer_fn, layer_map, tensor_map): argument
129 def _clone_functional_model(model, input_tensors=None, layer_fn=_clone_layer): argument
217 def _clone_layers_and_model_config(model, input_layers, layer_fn): argument
279 def _clone_sequential_model(model, input_tensors=None, layer_fn=_clone_layer): argument
|
| /aosp_15_r20/external/rust/android-crates-io/crates/tower-layer/src/ |
| D | lib.rs | 20 mod layer_fn; module
|
| D | layer_fn.rs | 67 pub fn layer_fn<T>(f: T) -> LayerFn<T> { in layer_fn() function
|
| /aosp_15_r20/external/rust/android-crates-io/crates/axum/src/routing/ |
| D | method_routing.rs | 932 let layer_fn = move |route: Route<B, E>| route.layer(layer.clone()); in layer() localVariable
974 let layer_fn = move |svc| { in route_layer() localVariable
|
| /aosp_15_r20/external/rust/android-crates-io/crates/tower/src/builder/ |
| D | mod.rs | 168 pub fn layer_fn<F>(self, f: F) -> ServiceBuilder<Stack<crate::layer::LayerFn<F>, L>> { in layer_fn() method
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 3141 layer_fn, argument
|