Searched refs:input_tangents (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 793 def forward(self, inference_args=None, input_tangents=None): argument
815 if input_tangents:
830 def record(self, flat_outputs, inference_args, input_tangents): argument
847 to_record, inference_args + input_tangents,
899 self, outputs, inference_args, input_tangents): argument
932 if input_tangents:
964 if not input_tangents:
971 forward_function, backward_function, inference_args, input_tangents)
982 if (len(inference_args) + len(input_tangents)
988 len(inference_args) + len(input_tangents),
[all …]
|
| D | pywrap_tfe_src.cc | 2399 PyObject* TangentsAsPyTuple(const std::vector<PyObject*>& input_tangents) { argument
2400 PyObject* py_input_tangents = PyTuple_New(input_tangents.size());
2401 for (int i = 0; i < input_tangents.size(); ++i) {
2403 if (input_tangents[i] == nullptr) {
2406 element = input_tangents[i];
2448 const std::vector<PyObject*>& input_tangents, argument
2456 TangentsAsPyTuple(input_tangents));
2478 const std::vector<PyObject*>& input_tangents, argument
2481 TangentsAsPyTuple(input_tangents));
2586 [forward_function](const std::vector<PyObject*>& input_tangents,
[all …]
|