- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for getInitValues (0.1 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
[](Type ty) { return mlir::cast<ShapedType>(ty).getElementType(); })}; // Create the mhlo.reduce op. auto reduce_op = rewriter.create<mhlo::ReduceOp>( loc, op.getInputs(), op.getInitValues(), GetI64ElementsAttr(op.getDimensionsToReduce()), elementTypes); // Insert a call to the reducer in the region of the mhlo op. BuildBodyWithCall(rewriter, loc, func, func_ty, &reduce_op.getBody());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)