Searched refs:input_tangents (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 698 def forward(self, inference_args=None, input_tangents=None): argument
720 if input_tangents:
734 def record(self, flat_outputs, inference_args, input_tangents): argument
751 to_record, inference_args + input_tangents,
803 self, outputs, inference_args, input_tangents): argument
836 if input_tangents:
868 if not input_tangents:
875 forward_function, backward_function, inference_args, input_tangents)
886 if (len(inference_args) + len(input_tangents)
899 inference_args, input_tangents): argument
[all …]
|
| D | pywrap_tfe_src.cc | 2471 PyObject* TangentsAsPyTuple(const std::vector<PyObject*>& input_tangents) { argument
2472 PyObject* py_input_tangents = PyTuple_New(input_tangents.size());
2473 for (int i = 0; i < input_tangents.size(); ++i) {
2475 if (input_tangents[i] == nullptr) {
2478 element = input_tangents[i];
2520 const std::vector<PyObject*>& input_tangents, argument
2528 TangentsAsPyTuple(input_tangents));
2550 const std::vector<PyObject*>& input_tangents, argument
2553 TangentsAsPyTuple(input_tangents));
2658 [forward_function](const std::vector<PyObject*>& input_tangents,
[all …]
|