- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 93 for mhlo (0.07 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
#include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/client/xla_builder.h" #include "xla/client/xla_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/service/hlo.pb.h" #include "xla/translate/hlo_to_mhlo/hlo_function_importer.h" #include "xla/translate/hlo_to_mhlo/hlo_to_mlir_hlo.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
} Value new_const_op = rewriter.create<mhlo::ConstantOp>(bcast_op.getLoc(), result); rewriter.replaceOp(bcast_op, {new_const_op}); return success(); } }; using FoldBroadcastInDimBeforeMulOp = FoldBroadcastInDimBeforeBinaryElementwiseOp<mhlo::MulOp>; // Constant folds mhlo.mul, this folder doesn't have an upper limit on how many // elements can be folded.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
if (lower_to_xla_hlo) { // This pass operates on MHLO control flow ops so it should be legalized // after the control flow ops are legalized. pm.addPass(mlir::mhlo::CreateLegalizeTFCommunicationPass()); // Everything should be MHLO after this. if (!allow_partial_conversion) { pm.addNestedPass<mlir::func::FuncOp>( mlir::mhlo::CreateVerifyTFXLALegalizationPass(legalize_chlo)); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.cc
// Legalize MHLO to StableHLO should be moved closer to where it is needed // There are some entry points that start with HLO->MHLO like // jax_to_tfl_flatbuffer.cc which can likely be updated to emit StableHLO // to be consistent with other entrypoints. pass_manager.addPass(mlir::mhlo::createHloLegalizeToStablehloPass()); // Decompose CHLO into StableHLO ops
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 25.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/hlo_matchers.cc
auto iota_op = dyn_cast_or_null<mhlo::IotaOp>(iota.getDefiningOp()); if (!iota_op || dimensions.getNumElements() != 1) return false; auto dim = *dimensions.value_begin<APInt>(); return dim == iota_op.getIotaDimension(); } // It matches %iota generated from the following mlir codes: // // %iota_r1 = mhlo.constant dense<[0, 1, 2, ..., L]> // %iota = "mhlo.broadcast_in_dim(%iota_r1){
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_patterns.td
HasMhloCompareType<"::mlir::mhlo::ComparisonType::UNSIGNED">, HasMhloCompareType<"::mlir::mhlo::ComparisonType::NOTYPE">]>, "compare type supported by TensorFlow">; class MHLO_ComparisonDirectionValue<string enumStr> : ConstantAttr<MHLO_ComparisonDirectionAttr, "::mlir::mhlo::ComparisonDirection::" # enumStr>; foreach p = [[TF_EqualOp, CHLO_ComparisonDirectionValue<"EQ">],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 03 08:58:22 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
// scatter like ops. template <> LogicalResult MatchBinaryReduceFunction<void>(mlir::Region& function) { Block& body = function.front(); if (body.getNumArguments() != 2) return failure(); mhlo::ReturnOp return_op = dyn_cast<mhlo::ReturnOp>(body.back()); if (!return_op) return failure(); if (return_op.getNumOperands() != 1) return failure(); if (return_op.getOperands().front() != body.getArgument(1)) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc
#include "tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_targets.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/mlir_hlo/mhlo/transforms/rewriters.h" #include "xla/mlir_hlo/mhlo/utils/type_conversion.h" #include "tensorflow/core/lib/monitoring/counter.h" namespace mlir { namespace mhlo { namespace { #define GEN_PASS_DEF_LEGALIZETF
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/stablehlo/odml_to_stablehlo.cc
mlir::registerAsmPrinterCLOptions(); mlir::registerAllPasses(); mlir::registerTensorFlowPasses(); mlir::mhlo::registerAllMhloPasses(); // These are in compiler/mlir/tf2xla and not part of the above MHLO passes. mlir::mhlo::registerTfXlaPasses(); mlir::mhlo::registerLegalizeTFPass(); mlir::xla_framework::registerXlaFrameworkPasses(); tensorflow::RegisterConvertMlirToXlaHloPipelineWithDefaults();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
custom_legalization_passes); // Compiles a Graph from TF to HLO and adds the resulting HLO to the // XlaBuilder. This function adds HLO to a larger HLO computation, so // HLO-level inputs are supplied, and HLO-level outputs are produced. // xla_params is the HLO-level inputs and returns is the HLO-level outputs. // If unconditionally_use_output_shapes is true then the unregistered
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 10.4K bytes - Viewed (0)