Searched defs:output_subs (Results 1 – 2 of 2) sorted by relevance
185 absl::string_view output_subs, in EinsumGradReducedHelper()284 absl::string_view output_subs) { in EinsumGradWrt()376 const absl::string_view output_subs = equation_split[1]; in EinsumGrad() local
143 def _GetGradReduced(output_grad, output_subs, input_subs, input_shape, argument221 other_subs, output_subs): argument