- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for getKeepDims (0.54 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
auto value_to_replace = avg_pool.getResult(); // If it's not keep dim, we need to insert a reshape after the average // pool. if (!mean_op.getKeepDims()) { // Insert the reshape. SmallVector<int64_t, 2> new_shape({batch, channel}); auto reshape_op = InsertReshapeOp(mean_op.getLoc(), avg_pool.getResult(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
if (!log_op || !log_op->hasOneUse()) { return failure(); } auto sum_op = dyn_cast_or_null<TFL::SumOp>(log_op.getX().getDefiningOp()); if (!sum_op || !sum_op.getKeepDims() || !isSupportedAxis( sum_op.getAxes(), mlir::cast<ShapedType>(sum_op.getOperand(0).getType()).getRank())) { return failure(); } if (!sum_op->hasOneUse()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
if (!result_ty) return {}; // Bypass this op if the result has the same shape and type. This can happen // if the input tensor has size 0 or size 1. if (!getKeepDims() && input_ty == result_ty) { return getInput(); } return {}; } //===----------------------------------------------------------------------===// // StridedSliceOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// reshape. Various code generation techniques benefit from the knowledge // that this is a restricted form of shape manipulation that is just adding // unit dims. if (op.getKeepDims()) { for (const auto &dim_is_reduced : llvm::enumerate(reduced_dimensions_bitmap)) { if (dim_is_reduced.value()) { auto index_attr = GetI32ElementsAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)