- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for FoldConstantTransposePass (0.23 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
}; }; } // namespace class FoldConstantTransposePass : public impl::FoldConstantTransposePassBase<FoldConstantTransposePass> { public: using impl::FoldConstantTransposePassBase< FoldConstantTransposePass>::FoldConstantTransposePassBase; private: void runOnOperation() override; }; void FoldConstantTransposePass::runOnOperation() { func::FuncOp func_op = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/passes.td
"mlir::stablehlo::StablehloDialect", "TF::TensorFlowDialect", "mlir::quant::QuantizationDialect", "mlir::quantfork::QuantizationForkDialect", ]; } def FoldConstantTransposePass : Pass<"stablehlo-fold-constant-transpose", "mlir::func::FuncOp"> { let summary = "Folds stablehlo.constant -> stablehlo.transpose patterns."; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/pipelines/process_nchw_tensor.mlir
// to match the function signature. Constants are also transpose-folded // accordingly. // // Note that the `transpose` after the `broadcast_in_dim` is not folded by the // `FoldConstantTransposePass`. // CHECK-LABEL: nchw_conv_with_broadcasted_bias_add_relu // CHECK-SAME: %[[ARG:.+]]: tensor<1x2x5x5xf32> func.func @nchw_conv_with_broadcasted_bias_add_relu(%arg0: tensor<1x2x5x5xf32>) -> tensor<1x4x5x5xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 20:32:46 UTC 2024 - 12.6K bytes - Viewed (0)