- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for UnfoldSplatConstantPass (0.2 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfold_splat_constant_pass.cc
// TODO(b/295966255): Remove this pass after moving MHLO folders to a separate // pass and folders are not applied by default. class UnfoldSplatConstantPass : public impl::UnfoldSplatConstantPassBase<UnfoldSplatConstantPass> { public: void runOnOperation() override { auto module = getOperation(); mlir::OpBuilder op_builder(&module.getBodyRegion());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/passes.td
let summary = "Legalize StableHLO custom call ops where the call target is 'stablehlo.composite' to composite ops."; let dependentDialects = ["mlir::stablehlo::StablehloDialect"]; } def UnfoldSplatConstantPass : Pass<"unfold-splat-constant-pass", "ModuleOp"> { let summary = "Replaces a splat constant tensor with a BroadcastInDim op."; let constructor = "mlir::odml::CreateUnfoldSplatConstantPass()"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 5.6K bytes - Viewed (0)