- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for legalization (0.24 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
return op->getResult(0).getType().cast<TensorType>().clone( new_result_quantized_type); } // Matches kernel dimension numbers, ranks of input and output and constant // kernel for legalization to TFLite convolution ops. LogicalResult MatchConvolutionFormat(stablehlo::ConvolutionOp op) { stablehlo::ConvDimensionNumbersAttr dimension_numbers = op.getDimensionNumbers();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
#define GEN_PASS_DEF_LEGALIZEHLOTOTFPASS #include "tensorflow/compiler/mlir/lite/stablehlo/transforms/passes.h.inc" class LegalizeHloToTf : public impl::LegalizeHloToTfPassBase<LegalizeHloToTf> { /// Performs the legalization to the TF dialect. void runOnOperation() override; }; using mhlo::DotDimensionNumbersAttr; // Replaces `region`'s terminator to TF::Yield. void ReplaceReturnOp(Region& region, PatternRewriter& rewriter) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
first = false; os << result_type; } } os << ")"; // Print out attributes except for large elementsattributes (which should // rarely be the cause why the legalization didn't happen). if (!inst->getAttrDictionary().empty()) { os << " : {"; bool first = true; for (auto& named_attr : inst->getAttrDictionary()) { os << (!first ? ", " : ""); first = false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
let hasFolder = 1; } // If there is a change in supporting more types in the TFLite cast op kernel, // the While loop outline pass should be updated since it inserts cast op(s) // after the TF -> TFL legalization pass is done. // LINT.IfChange def TFL_CastOp : TFL_Op<"cast", [ Pure, SameOperandsAndResultShape]> { let summary = "Cast operator"; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
func.return %0 : tensor<16xf32> } // ----- // Test tf.Min with complex numbers. // Previous versions of tensorflow said complex numbers were allowed with // tf.Min even though it doesn't make sense. The legalization of tf to xla // requires that complex types are not allowed in tf.Min, so we have an // explicit unit here to make sure that invariant is enforced.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0)