- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 124 for legalized (0.14 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc
EXPECT_FALSE(IsDynamicPadderOp(TypeID::get<TF::ConstOp>())); } // This test is kind of odd. We go through all the Tensorflow types and check // whether they are legalized with MLIR, TF2XLA, or both. Ideally the sets are // disjoint, but until that happens, this tests ensures that the set doesn't // grow. TEST_F(LegalizationOpConfigTest, CountLoweringsSet) { int mlir_lowering_count = 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 03:31:01 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
} return success(); } virtual bool IsValueInitValue(const DenseElementsAttr& attr) const = 0; }; // Returns true if the given reduce op can be legalized to ArgMax/ArgMin ops. std::optional<bool> IsReduceOpLegal(mhlo::ReduceOp reduce_op); } // namespace odml } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/split_merged_operands.cc
if (inserted_value) continue; // We can only clone the constant op or const->dequantize combo. The latter // case is useful for float16 quantization. Since all ops have been // legalized to tflite ops, so we only care about ConstOp or QConstOp or // mlir constant op. Operation* input_op = operand.getDefiningOp(); if (input_op == nullptr) return failure(); Attribute attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::ModOp>(), // MatrixDiagPartV3 should use the MLIR implementation due to performance. TypeID::get<TF::MatrixDiagPartV3Op>(), // Ops that are legalized in the old bridge using MlirXlaOpKernel TypeID::get<TF::AbsOp>(), TypeID::get<TF::AtanOp>(), TypeID::get<TF::AvgPool3DOp>(), TypeID::get<TF::BiasAddGradOp>(), TypeID::get<TF::CeilOp>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
Value output_lhs = bmm_op.getAdjX() ? create_z_x_transpose_op(input_lhs) : input_lhs; // The rhs need to be transposed if adj_y == false AND this matmul will be // legalized to tfl.fully_connected Value output_rhs = !bmm_op.getAdjY() ? create_z_x_transpose_op(input_rhs) : input_rhs; Type output_type = bmm_op.getResult().getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K 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/tf2xla/transforms/xla_legalize_tf.cc
"/tensorflow/core/tf2xla/v1/mlir_failed_xla_legalize_tf_pass_count", "Counts the failure of legalization of ops", "op_name", "legality"); class LegalizeTF : public impl::LegalizeTFBase<LegalizeTF> { public: explicit LegalizeTF(bool legalize_chlo, std::optional<StringRef> tf2xla_fallback_device_type, bool prefer_tf2xla) { legalize_chlo_ = legalize_chlo;
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/transforms/legalize_hlo_patterns.td
def : Pat<(MHLO_ConstantOp:$output $value), (TF_ConstOp $value), [(TF_Tensor $output)]>; //===----------------------------------------------------------------------===// // Binary op patterns. // Note that these are legalized from chlo.broadcast_* ops, since those are // semantically compatible with the corresponding TF ops. Depending on // context, getting to these ops may require some raising.
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/transforms/legalize_tf.cc
// This transformation pass converts operations in TensorFlow dialect into // operations that are legal in the TensorFlow Lite dialect. Operations that // can be legalized to TensorFlow Lite dialect with simple replacements are part // of this pass and other operations that may create extra ops should be part of // the PrepareTF pass which should be run before this pass. That way any
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
lower_to_xla_hlo); 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>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0)