Searched defs:input_subs (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | linalg_grad.cc | 186 absl::string_view input_subs, in EinsumGradReducedHelper()
283 absl::string_view input_subs, absl::string_view other_subs, in EinsumGradWrt()
375 const absl::string_view input_subs = equation_split[0]; in EinsumGrad() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | linalg_grad.py | 143 def _GetGradReduced(output_grad, output_subs, input_subs, input_shape, argument
220 def _GetGradWrt(output_grad, other_operand, input_shape, input_subs, argument
|