Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for EinsumGradWrt (0.12 sec)

  1. tensorflow/cc/gradients/linalg_grad.cc

        x = Conj(scope, x);
        y = Conj(scope, y);
      }
    
      const auto x_shape = Shape(scope, x);
      const auto y_shape = Shape(scope, y);
      Output grad_x =
          EinsumGradWrt(scope, grad, y, x_shape, x_subs, y_subs, output_subs);
      Output grad_y =
          EinsumGradWrt(scope, grad, x, y_shape, y_subs, x_subs, output_subs);
    
      if (!absl::StrContains(output_subs, kEllipsis)) {
        // If no ellipsis in the output; then no need to unbroadcast.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 07 23:11:54 UTC 2022
    - 20.4K bytes
    - Viewed (0)
Back to top