- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for PopulateLegalizeTfWithTf2XlaPatterns (0.42 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla_passes.h
/// `prefer_tf2xla` means an op will be included iff it is not in /// `MlirLegalizedUnderPreferTf2XlaSet`. `!prefer_tf2xla` mean an op will be /// included if there is no native MLIR legalization for the op. void PopulateLegalizeTfWithTf2XlaPatterns(llvm::StringRef device_type, RewritePatternSet& patterns, MLIRContext* ctx,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
RewritePatternSet patterns(context); mhlo::PopulateLegalizeTfPatterns(context, &patterns); TF::PopulateTFLoweringBeforeHLOPatterns(context, &patterns); mhlo::Tf2XlaTypeConverter converter; mhlo::PopulateLegalizeTfWithTf2XlaPatterns( "XLA_CPU_JIT", patterns, context, converter, /*prefer_tf2xla=*/false); stablehlo::StablehloToHloTypeConverter hlo_converter; chlo::populateChloToHloPatterns(context, &hlo_converter, &patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc
return builder.create<mlir::tensor::CastOp>(loc, result_type, inputs.front()); }; addSourceMaterialization(cast_value); } void PopulateLegalizeTfWithTf2XlaPatterns(llvm::StringRef device_type, RewritePatternSet& patterns, MLIRContext* ctx,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
target.addIllegalOp<TF::XlaConvV2Op>(); target.addIllegalOp<TF::XlaGatherOp>(); RewritePatternSet patterns(context); mhlo::Tf2XlaTypeConverter converter; mhlo::PopulateLegalizeTfWithTf2XlaPatterns("XLA_CPU_JIT", patterns, context, converter); mhlo::PopulateLegalizeTfPatterns(context, &patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0)