- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for Tf2XlaTypeConverter (0.24 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla_passes.h
class Operation; template <typename T> class OperationPass; class Pass; namespace mhlo { /// Converter to be used along with the fallback Tf2Xla patterns below. class Tf2XlaTypeConverter : public TypeConverter { public: Tf2XlaTypeConverter(); }; /// Adds the TF to XLA via TF2XLA rewrite patterns to the pattern list. /// `prefer_tf2xla` means an op will be included iff it is not in
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/tf2xla/transforms/legalize_tf_with_tf2xla.cc
Operation* cloned_op = rewriter.clone(*op, mapper); rewriter.replaceOp(op, cloned_op->getResults()); return success(); } }; } // end namespace Tf2XlaTypeConverter::Tf2XlaTypeConverter() { // Currently, we don't do any type conversions. Any TensorFlow op with a type // that is not supported in MHLO will fail conversion. Quantized types are
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/lite/stablehlo/transforms/tf_stablehlo_pass.cc
MLIRContext *context = func->getContext(); 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;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0)