Searched refs:forward_output (Results 1 – 5 of 5) sorted by relevance
26 forward_output = modeldef.forward(*modeldef.inputs)30 backward_input = modeldef.backward_setup(forward_output)32 backward_input = forward_output
99 forward_output = modeldef.forward(*modeldef.inputs)106 backward_input = modeldef.backward_setup(forward_output)108 backward_input = forward_output
43 forward_output = benchmark(cuda_sync, modeldef.forward, *modeldef.inputs)
233 def varlen_lstm_backward_setup(forward_output, seed=None): argument237 sequences = forward_output[0]
810 Value* forward_output = forwardSV->asValue(loc, caller); in call() local821 NamedValue(forward_output)}, in call()826 forward_output = forward_hook_output; in call()831 return std::make_shared<SimpleValue>(forward_output); in call()