Searched refs:forward_function (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 547 forward_function = _EagerDefinedFunction(
550 return forward_function, backward_function
641 forward_function, backward_function = _create_forward_backward_with_graph(
643 return forward_function, backward_function
647 forward_function, backwards_function = self.forward_backward(len(doutputs))
650 forward_function.add_to_graph(op.graph)
654 op._set_func_attr("f", forward_function.name)
655 op._set_type_list_attr("Tout", forward_function._output_types)
657 forward_function._output_types[len(op.outputs):],
658 forward_function._output_shapes[len(op.outputs):])
[all …]
|
| D | tape.py | 178 forward_function=None): argument
182 forward_function)
|
| D | pywrap_tfe.h | 218 PyObject* forward_function);
|
| D | pywrap_tfe_src.cc | 2390 const tensorflow::eager::ForwardFunction<PyObject>* forward_function, argument
2454 forward_function, backward_function_getter, backward_function_killer);
2591 const tensorflow::eager::ForwardFunction<PyObject>* forward_function) { argument
2607 forward_function, nullptr /* No special-cased jvps. */,
2622 PyObject* forward_function) { argument
2649 if (forward_function == Py_None) {
2658 [forward_function](const std::vector<PyObject*>& input_tangents,
2661 return CallOpSpecificJVPFunction(forward_function, input_tangents,
3290 tensorflow::eager::ForwardFunction<PyObject>* forward_function; local
3300 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 | 823 forward_function=lambda x: [x])
997 forward_function=lambda x: [x])
1016 forward_function=lambda x: [x])
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 1066 const py::handle& forward_function) { in PYBIND11_MODULE() argument
1069 backward_function.ptr(), forward_function.ptr())); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 714 forward_function=lambda x: [x])
|