Home
last modified time | relevance | path

Searched defs:forward_pass (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
H A Dvariable_scope_shim_test.py869 def forward_pass(self, inputs, training=None): member in TF1VariableScopeWrapperLayerTest.test_get_variable.WrappedDenseLayer
929 def forward_pass(self, inputs, training=None): member in TF1VariableScopeWrapperLayerTest.test_compat_v1_layer.WrappedDenseLayer
H A Dvariable_scope_shim.py739 def forward_pass(self, *args, **kwargs): member in VariableScopeWrapperLayer
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dtimm_models.py396 def forward_pass(self, mod, inputs, collect_outputs=True): member in TimmRunner
H A Dhuggingface.py516 def forward_pass(self, mod, inputs, collect_outputs=True): member in HuggingfaceRunner
H A Dtorchbench.py436 def forward_pass(self, mod, inputs, collect_outputs=True): member in TorchBenchmarkRunner
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction_gradients_test.py931 def forward_pass(value): function