Searched refs:attr_tuple (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop.py | 47 def _identity_jvp(attr_tuple, inputs, outputs, tangents): argument
51 del attr_tuple, inputs, outputs
58 def _read_variable_jvp(attr_tuple, inputs, outputs, tangents): argument
61 del attr_tuple, inputs, outputs
74 def _jvp_helper(op_name, attr_tuple, inputs, outputs, tangents): argument
99 return special_case(attr_tuple, inputs, outputs, tangents)
119 execute.record_gradient(op_name, inputs, attr_tuple, outputs)
146 def _jvp_helper_wrapper(op_name, attr_tuple, inputs, outputs, tangents, argument
175 functools.partial(_jvp_helper, op_name, attr_tuple, inputs, outputs),
178 return _jvp_helper(op_name, attr_tuple, inputs, outputs, tangents)
[all …]
|
| D | forwardprop_test.py | 233 attr_tuple=(),
245 attr_tuple=(),
258 attr_tuple=(),
275 attr_tuple=(),
292 attr_tuple=(),
|
| D | backprop.py | 129 def _gradient_function(op_name, attr_tuple, num_inputs, inputs, outputs, argument
147 mock_op = _MockOp(attr_tuple, inputs, outputs, op_name, skip_input_indices)
|