- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 81 for legalize (0.36 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-collective.mlir
// RUN: tf-opt -split-input-file -verify-diagnostics -xla-legalize-tf-collective -xla-legalize-tf %s | FileCheck %s // CHECK: module attributes // CHECK-SAME{LITERAL}: tf2xla.collective_info.group_key = 0 // CHECK-SAME{LITERAL}: tf2xla.collective_info.group_size = 2 // CHECK-LABEL: func @all_reduce_cross_replica func.func @all_reduce_cross_replica(%input: tensor<f32>) -> tensor<f32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir
// RUN: tf-opt "-xla-legalize-tf=device-type=XLA_CPU_JIT legalize-chlo=false use-tf2xla-fallback=true prefer-tf2xla=true" %s | FileCheck %s // RUN: tf-opt "-xla-legalize-tf=device-type=XLA_CPU_JIT legalize-chlo=false prefer-tf2xla=true" %s | FileCheck --check-prefix NOFALLBACK %s module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}} { // CHECK-LABEL: @abs func.func @abs(%arg0: tensor<2xf32>) -> tensor<2xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-binary-elementwise.mlir
// (unlike the rest), since this is the primary use case for such ops and // verification of shapes and broadcasts is desired. // RUN: tf-opt "-xla-legalize-tf=legalize-chlo=true" -canonicalize %s | FileCheck %s // RUN: tf-opt "-xla-legalize-tf=legalize-chlo=false" %s | FileCheck --check-prefix CHLO %s //===----------------------------------------------------------------------===// // Binary op legalizations.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 18.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-quant.mlir
// RUN: tf-opt -convert-tf-quant-types -quant-convert-tf-quant-ops-to-mhlo -canonicalize "-xla-legalize-tf=legalize-chlo=false" -split-input-file %s | FILECHECK_OPTS="" FileCheck %s //===----------------------------------------------------------------------===// // tf.UniformQuantizedDotHybrid legalization //===----------------------------------------------------------------------===// // CHECK-LABEL: func @quantized_matmul_fn
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 01:25:29 UTC 2024 - 37.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-communication.mlir
// RUN: tf-opt -split-input-file -verify-diagnostics -xla-legalize-tf-communication %s | FileCheck %s // Test legalization of `tf._XlaHostComputeMlir` expands into individual // `mhlo.send` per operand and `mhlo.recv` per result. Channel Id's are uniquely // assigned per mhlo communcation op, and frontend attributes (modified keys) // and op shardings are added. Sink tokens are created // if there are more than one operand or more than one result. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 08 18:24:20 UTC 2024 - 38.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir
// RUN: tf-opt "-xla-legalize-tf=device-type=XLA_CPU_JIT prefer-tf2xla=true use-tf2xla-fallback=true" %s -verify-diagnostics -mlir-disable-threading | FileCheck %s module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}} { // CHECK-LABEL: binary_op func.func @binary_op(%arg0: tensor<2xf32>, %arg1: tensor<2xf32>) -> tensor<2xf32> { // CHECK: mhlo.atan2
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.h
// tensor to sparse format. std::unique_ptr<OperationPass<func::FuncOp>> CreateDenseToSparsePass(); // Creates function pass to legalize TF While to TFL While. std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFWhilePass(); // Legalize tflite flex ops to TF ops. std::unique_ptr<OperationPass<func::FuncOp>> CreateLiftTfliteFlexOpsPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 07 21:29:34 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.td
]; } def LegalizeWhilePass : Pass<"tfl-legalize-tf-while", "mlir::ModuleOp"> { let summary = "Legalize from TensorFlow While to TensorFlow Lite While."; let constructor = "CreateLegalizeTFWhilePass()"; let dependentDialects = ["TFL::TensorFlowLiteDialect"]; } def LegalizeVariablesPass : Pass<"tfl-legalize-variables-tf", "mlir::ModuleOp"> { let summary = "Legalize TensorFlow variables to TensorFlow Lite dialect.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 22.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.cc
pass_manager->addPass(mlir::TF::CreateTFShapeInferencePass()); } // Legalize while early to allow further constant folding. // TODO(jpienaar): This may not actually matter as we do canonicalization // after the legalize below, for now it needs to be below the above passes // that work on TF dialect and before inliner so that the function calls in
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/legalize_stablehlo_to_vhlo.cc
for (auto arg : block.getArguments()) { ConvertAndWrapUsesInUnrealizedCast(arg, converter, rewriter); } } } }); }); return success(); } // Legalize StableHLO portion of program to VHLO, leaves TFL untouched LogicalResult ApplyStablehloToVhloPatterns(ModuleOp module, bool is_func_legal) { MLIRContext *context = module.getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 19:48:51 UTC 2024 - 12.3K bytes - Viewed (0)