Searched refs:forward_function (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 642 forward_function = _EagerDefinedFunction(
645 return forward_function, backward_function
736 forward_function, backward_function = _create_forward_backward_with_graph(
738 return forward_function, backward_function
742 forward_function, backwards_function = self.forward_backward(len(doutputs))
745 forward_function.add_to_graph(op.graph)
749 op._set_func_attr("f", forward_function.name)
750 op._set_type_list_attr("Tout", forward_function._output_types)
752 forward_function._output_types[len(op.outputs):],
753 forward_function._output_shapes[len(op.outputs):])
[all …]
|
| D | tape.py | 182 forward_function=None): argument
186 forward_function)
|
| D | pywrap_tfe.h | 214 PyObject* forward_function);
|
| D | pywrap_tfe_src.cc | 2318 const tensorflow::eager::ForwardFunction<PyObject>* forward_function, argument
2382 forward_function, backward_function_getter, backward_function_killer);
2519 const tensorflow::eager::ForwardFunction<PyObject>* forward_function) { argument
2535 forward_function, nullptr /* No special-cased jvps. */,
2550 PyObject* forward_function) { argument
2577 if (forward_function == Py_None) {
2586 [forward_function](const std::vector<PyObject*>& input_tangents,
2589 return CallOpSpecificJVPFunction(forward_function, input_tangents,
3211 tensorflow::eager::ForwardFunction<PyObject>* forward_function; local
3221 forward_function = nullptr;
[all …]
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 286 const ForwardFunction<Gradient>* forward_function,
1027 const ForwardFunction<Gradient>* forward_function, in Accumulate() argument
1085 if (forward_function == nullptr) { in Accumulate()
1094 (*forward_function)(in_grads, &forward_grads, use_batch_)); in Accumulate()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | func_graph.py | 668 forward_function=lambda x: [x])
719 forward_function=lambda x: [x])
740 forward_function=lambda x: [x])
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 907 const py::handle& forward_function) { in PYBIND11_MODULE() argument
910 backward_function.ptr(), forward_function.ptr())); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 683 forward_function=lambda x: [x])
|