- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 173 for mhlo (0.16 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/verify_tfxla_legalization_test.cc
%0 = mhlo.constant dense<1.000000e+00> : tensor<f64> %1 = mhlo.convert %0 : (tensor<f64>) -> tensor<i64> %2 = mhlo.reshape %1 : (tensor<i64>) -> tensor<1xi64> %3 = "mhlo.dynamic_iota"(%2) {iota_dimension = 0 : i64} : (tensor<1xi64>) -> tensor<?xi32> %4 = mhlo.multiply %3, %3 : tensor<?xi32> return %4 : tensor<?xi32> } })";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.h
// Compiles the given Operation with XlaBuilder and imports the generated HLO // via the HLO -> MHLO importer. absl::StatusOr<mhlo::TupleOp> CompileWithHloImporter( tensorflow::OpKernelContext& op_context); // Import the given XlaComputation into the parent module. Returns the given // generated function. absl::StatusOr<mhlo::TupleOp> ImportXlaComputation( xla::XlaComputation& computation);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/verify-tfxla-legalization.mlir
%0 = mhlo.constant dense<1.000000e+00> : tensor<f64> %1 = mhlo.convert %0 : (tensor<f64>) -> tensor<i64> %2 = mhlo.reshape %1 : (tensor<i64>) -> tensor<1xi64> // expected-error @+1 {{Node `mhlo.dynamic_iota` must have compile-time constant}} %3 = "mhlo.dynamic_iota"(%2) <{iota_dimension = 0 : i64}> : (tensor<1xi64>) -> tensor<?xi32> %4 = mhlo.multiply %3, %3 : tensor<?xi32> return %4 : tensor<?xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.h
namespace mlir { namespace odml { // Converts mhlo.dot_general to tfl.BatchMatMul. Reshape and Transpose ops will // be inserted to convert to well-formed matrix multiply; i.e., mhlo.dot_general // -> tfl.batch_matmul(mhlo.transpose(mhlo.reshape(operand)), ...). // Note: // 1) Reshape/transpose are inserted because tfl.BatchMatMul requires // size(contracting_dimensions) = 1 and size(output_dim) = 1, whereas // mhlo.dot_general has no such restriction.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 04 19:00:01 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/passes.h
namespace mlir { namespace odml { // Creates a pass which unfuses MHLO batch norm inference op into arithmetic // ops. std::unique_ptr<Pass> createUnfuseBatchNormPass(); // Creates a pass which constant folds broadcast_in_dim op conditionally. std::unique_ptr<Pass> createFoldBroadcastPass(); // Creates a pass which fuses MHLO binary element-wise ops and convolution op. std::unique_ptr<Pass> createFuseConvolutionPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/order_by_dialect.mlir
return %14 : tensor<32x10xf32> } // CHECK: ReadVariableOp // CHECK: mhlo.convolution // CHECK: mhlo.add // CHECK: mhlo.maximum // CHECK: mhlo.reshape // CHECK: mhlo.dot // CHECK: mhlo.add // CHECK: mhlo.maximum // CHECK: return{{.*}}tensor<32x10xf32> // ----- // CHECK-LABEL: @mhlo_while func.func private @mhlo_while() { // CHECK-NEXT: mhlo.constant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-include-tf2xla-fallback.mlir
func.func @batchmatmulv2(%arg0: tensor<1x4x2xf32>, %arg1: tensor<3x2x4xf32>) -> tensor<3x4x4xf32> { // NO_FALLBACK: mhlo.dynamic_broadcast_in_dim // NO_FALLBACK: mhlo.dot_general // SUPPORTED_FALLBACK_DEVICE: mhlo.reduce // SUPPORTED_FALLBACK_DEVICE: mhlo.dot_general // SUPPORTED_FALLBACK_DEVICE: mhlo.transpose
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 16 19:04:03 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/passes.td
let summary = "Convert TF Quant ops to MHLO quantizated ops."; let description = [{ Convert TF Quant ops to MHLO quant ops. }]; let constructor = "mlir::quant::stablehlo::CreateConvertTFQuantOpsToMHLOPass()"; let dependentDialects = ["TF::TensorFlowDialect", "chlo::ChloDialect", "mhlo::MhloDialect", "tf_type::TFTypeDialect",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 23 01:41:18 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.cc
return failure(); auto compare_direction = is_argmax ? mhlo::ComparisonDirection::GT : mhlo::ComparisonDirection::LT; if (is_float) { mhlo::OrOp value_or = llvm::dyn_cast_or_null<mhlo::OrOp>( value_select.getOperand(0).getDefiningOp()); if (!value_or) return failure(); mhlo::CompareOp value_gt = llvm::dyn_cast_or_null<mhlo::CompareOp>( value_or.getLhs().getDefiningOp());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/group_by_dialect.mlir
// CHECK: func @three_dialects // CHECK-NEXT: glue.constant // CHECK-NEXT: glue.compare // CHECK-NEXT: call{{.*}}mhlo // CHECK-NEXT: call{{.*}}tf // CHECK-NEXT: return // CHECK: func.func{{.*}}dialect = "mhlo" // CHECK-NEXT: mhlo.constant // CHECK-NEXT: mhlo.multiply // CHECK: func.func{{.*}}dialect = "tf" // CHECK-NEXT: tf.Const // CHECK-NEXT: tf.AddV2 // CHECK-NEXT: tf.Identity
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 28 23:43:21 UTC 2022 - 5.7K bytes - Viewed (0)