- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for EinsumGradWrt (0.12 sec)
-
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)