- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for ReduceWindowOp (0.34 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
// `stablehlo.maximum`, representing max pooling. class DeferActivationTransposeForMaxPoolReduceWindowOp : public OpRewritePattern<mlir::stablehlo::ReduceWindowOp> { public: using OpRewritePattern<mlir::stablehlo::ReduceWindowOp>::OpRewritePattern; LogicalResult match(mlir::stablehlo::ReduceWindowOp op) const override { if (failed(MatchMaxPoolReduceWindowOp(op))) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
// Transform reduce_window(transpose(x)) to transpose(reduce_window(x)) struct TransposeCommuteWithReduceWindow : public OpRewritePattern<stablehlo::ReduceWindowOp> { using OpRewritePattern<stablehlo::ReduceWindowOp>::OpRewritePattern; LogicalResult matchAndRewrite(stablehlo::ReduceWindowOp reduce_op, PatternRewriter& rewriter) const override { MLIRContext* ctx = reduce_op.getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec.cc
mlir::stablehlo::DynamicReshapeOp, mlir::stablehlo::DynamicSliceOp, mlir::stablehlo::GatherOp, mlir::stablehlo::PadOp, mlir::stablehlo::ReduceWindowOp, mlir::stablehlo::ReshapeOp, mlir::stablehlo::SelectOp, mlir::stablehlo::SliceOp, mlir::stablehlo::TransposeOp>(op)) { scale_spec->has_same_scale_requirement = true; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
!IsConnectedWithQuantizedCompsiteFunction(candidate_op)) { return failure(); } // Ops with regions will be quantized in a separate pattern. if (isa<mlir::stablehlo::ReduceWindowOp>(candidate_op)) { return failure(); } const bool weight_only_quantizable = IsWeightOnlyQuantizableOp(*candidate_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
} }; // A helper function for ConvertMaxPoolOp and ConvertAvgPoolOp. Returns true // if the given ReduceWindowOp is a spatial pooling without dilation. If returns // true, also outputs the window strides and the TF padding mode ("VALID" or // "SAME"). bool IsSpatialPoolingWithoutDilation( mhlo::ReduceWindowOp rw, llvm::SmallVectorImpl<int64_t>* window_strides, std::string* padding_mode, std::string* data_format) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// computations, compared to the `xla::AvgPoolGrad` implementation, by // subtracting the original exterior padding before `ReduceWindowOp` // instead of trimming the result of `ReduceWindowOp` (the final result is // the same because all strides are 1). low_padding[dim] = conv_grad_spatial_dim.pad_before - orig_padding[dim].first;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
} }; // Rewrites quantized stablehlo.reduce_window with max to tfl.max_pool_2d. class RewriteQuantizedReduceWindowOpWithMax : public OpRewritePattern<stablehlo::ReduceWindowOp> { public: using OpRewritePattern<stablehlo::ReduceWindowOp>::OpRewritePattern; LogicalResult MatchBinaryReduceFunction(Region& function) const { Block& body = function.front(); if (body.getNumArguments() != 2) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
return failure(); } for (Operation* op_with_region : op.getResult().getUsers()) { // Among the ops with regions, only reduce_window op is supported for now. if (!isa<mlir::stablehlo::ReduceWindowOp>(op_with_region)) { return failure(); } if (!IsNestedRegionQuantizable(op_with_region)) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
const std::vector<int32_t>& operands, const std::vector<int32_t>& results); std::optional<BufferOffset<tflite::Operator>> BuildStablehloReduceWindowOp( mlir::stablehlo::ReduceWindowOp reduce_window_op, const std::vector<int32_t>& operands, const std::vector<int32_t>& results); std::optional<BufferOffset<tflite::Operator>> BuildStablehloRngBitGeneratorOp(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)