- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for Contraction (0.81 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
if (!isa<func::FuncOp, IfOp, WhileOp>(contraction->getParentOp())) { return rewriter.notifyMatchFailure( contraction, "fused operation must be nested inside a function, If or While"); } // If the contraction is used in multiple places, fusing it will only create // more contraction nodes, which is slower. if (!contraction.getResult().hasOneUse())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/cc/gradients/linalg_grad.cc
absl::string_view output_subs) { // Claim: For the einsum operation z = einsum("{eq_x},{eq_y}->{eq_z}", x, y), // where the equation involves only Tensor contractions, generalized traces // and transposes, the input gradients are given by the vector-jacobian // products (VJPs): // // grad_wrt_x = einsum("{eq_y},{eq_z}->{eq_x}", y, grad_wrt_z)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 07 23:11:54 UTC 2022 - 20.4K bytes - Viewed (0)