- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 29 for MhloDialect (0.15 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/verify_tfxla_legalization.cc
if (!HasStaticShapeOrBounded(o)) { return EmitMustBeConstantError(op); } } return true; } bool IsMhloAndStatic(Operation* op) { if (!llvm::isa<mlir::mhlo::MhloDialect>(op->getDialect())) { // Skip this op if it isn't an mhlo op. return true; } return IsStaticOperation(op); } bool IsDefaultConversionLegal(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/utils/tf_type_utils_test.cc
context->appendDialectRegistry(mlir_registry); context->getOrLoadDialect<tf_type::TFTypeDialect>(); context->getOrLoadDialect<quant::QuantizationDialect>(); context->getOrLoadDialect<mlir::mhlo::MhloDialect>(); context->getOrLoadDialect<sparse_tensor::SparseTensorDialect>(); return context; } TEST(GetDenseAttrFromTensorProtoAttrTest, Qint8ToUQ8Succeeds) { auto context = CreateContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.h
namespace mlir { namespace quant { class QuantizationDialect; } namespace quantfork { class QuantizationForkDialect; } namespace mhlo { class MhloDialect; } namespace TF { class TensorFlowDialect; } namespace TFL { class TFLDialect; typedef TFLDialect TensorFlowLiteDialect; } // namespace TFL namespace func { class FuncOp; } class ModuleOp;
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/tensorflow/transforms/tf_passes.td
// multi-threaded execution context, and PassManager is // multi-threaded. let dependentDialects = [ "chlo::ChloDialect", "mhlo::MhloDialect", "shape::ShapeDialect", "stablehlo::StablehloDialect", "vhlo::VhloDialect", "quant::QuantizationDialect", ]; let constructor = "TF::CreateXlaCallModuleDeserializationPass()"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
MLIRContext *ctx = &getContext(); func::FuncOp func = getOperation(); ConversionTarget target(*ctx); target.addLegalDialect<TF::TensorFlowDialect, mhlo::MhloDialect, chlo::ChloDialect>(); target.addIllegalOp< TF::UniformQuantizeOp, TF::UniformRequantizeOp, TF::UniformDequantizeOp, TF::UniformQuantizedDotOp, TF::UniformQuantizedDotHybridOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.td
let constructor = "CreatePrepareTFPass()"; let dependentDialects = ["TFL::TensorFlowLiteDialect", "mlir::quant::QuantizationDialect", "mlir::quantfork::QuantizationForkDialect", "mhlo::MhloDialect" ]; let options = [ Option<"unfold_batch_matmul_", "unfold_batchmatmul", "bool", "true", "Unfold BatchMatMul into individual MatMul ops.">,
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/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
class ConvertTfQuantToMhloIntTest : public Test { protected: void SetUp() override { DialectRegistry dialects; dialects.insert<TF::TensorFlowDialect, func::FuncDialect, chlo::ChloDialect, mhlo::MhloDialect, quant::QuantizationDialect>(); ctx_ = std::make_unique<MLIRContext>(dialects); ctx_->loadAllAvailableDialects(); // Create a CPU client with 1 device. TF_ASSERT_OK_AND_ASSIGN(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/dialect_detection_utils_test.cc
namespace tf2xla { namespace internal { namespace { using mlir::MLIRContext; using mlir::OpBuilder; using mlir::Operation; using mlir::OperationState; using mlir::UnknownLoc; using mlir::chlo::ChloDialect; using mlir::TF::TensorFlowDialect; using tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects; class SharedUtilsTest : public ::testing::Test {}; TEST_F(SharedUtilsTest, IsInFunctionalDialectPasses) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_to_vhlo.cc
public: StablehloToOdmlTypeConverter() : vhlo::VhloTypeConverter() { addConversion([](Type type) { if (type.getDialect().getNamespace() == vhlo::VhloDialect::getDialectNamespace()) { return type; } LLVM_DEBUG(llvm::dbgs() << "Invalid type: " << type << '\n'); return Type(); }); addConversion([](stablehlo::TokenType token) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 19:48:51 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/passes.td
ops, but may not have identical structure. }]; let options = []; let constructor = "CreateOutlineCompositesPass()"; let dependentDialects = ["mlir::chlo::ChloDialect", "mlir::stablehlo::StablehloDialect", "mlir::func::FuncDialect"]; } def SHLOSimplifyPass: Pass<"shlo-simplify", "ModuleOp"> { let summary = "Apply internal canonicalizations and foldings."; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:05:30 UTC 2024 - 1.7K bytes - Viewed (0)