- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 36 for TensorFlowDialect (0.93 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_fake_quant_to_qdq.cc
return "Convert Fake Quant op to quant.qcast and quant.dcast pairs"; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); registry.insert<quant::QuantizationDialect>(); registry.insert<quantfork::QuantizationForkDialect>(); } void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 15 00:56:15 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf_mlir_reduce_main.cc
tensorflow::InitMlir y(&argc, &argv); mlir::DialectRegistry registry; mlir::RegisterCommonToolingDialects(registry); registry.addExtension( +[](mlir::MLIRContext *ctx, mlir::TF::TensorFlowDialect *dialect) { dialect->addInterfaces<TFReductionPatternInterface>(); }); mlir::MLIRContext context(registry); return failed(mlirReduceMain(argc, argv, context));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 21 20:13:57 UTC 2023 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/odml_converter_main.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:05:30 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/tfr_opt.cc
int main(int argc, char **argv) { tensorflow::InitMlir y(&argc, &argv); mlir::registerAllPasses(); mlir::DialectRegistry registry; registry.insert<mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, mlir::arith::ArithDialect, mlir::func::FuncDialect, mlir::shape::ShapeDialect, mlir::quant::QuantizationDialect, mlir::quantfork::QuantizationForkDialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 15 16:15:16 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc
TEST_F(LegalizationOpConfigTest, CountLoweringsSet) { int mlir_lowering_count = 0; int tf2xla_fallback_count = 0; int non_categorized_count = 0; DialectRegistry dialect_registry; dialect_registry.insert<mlir::TF::TensorFlowDialect>(); MLIRContext context(dialect_registry); context.loadAllAvailableDialects(); for (auto operation : context.getRegisteredOperations()) { if (IsTypeLegalizedWithMlir(operation.getTypeID())) {
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/register_common_dialects.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 22:45:35 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_opt.cc
tensorflow::InitMlir y(&argc, &argv); mlir::registerAllPasses(); mlir::registerTensorFlowPasses(); mlir::DialectRegistry registry; registry.insert<mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, mlir::tf_saved_model::TensorFlowSavedModelDialect, mlir::func::FuncDialect, mlir::shape::ShapeDialect, mlir::arith::ArithDialect, mlir::tf_type::TFTypeDialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:20:34 UTC 2023 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils_test.cc
MLIRContext context; OwningOpRef<ModuleOp> module(ModuleOp::create(UnknownLoc::get(&context))); OpBuilder builder(&module->getBodyRegion()); context.loadDialect<TF::TensorFlowDialect>(); Value value = CreateConstValue<int8_t>(builder, module->getLoc(), unpacked_shape, unpacked_values);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_targets_test.cc
registry.insert<arith::ArithDialect>(); registry.insert<func::FuncDialect>(); registry.insert<tensor::TensorDialect>(); registry.insert<shape::ShapeDialect>(); registry.insert<TF::TensorFlowDialect>(); registry.insert<chlo::ChloDialect>(); return registry; } class XlaLegalizeTargetsTest : public testing::Test { public: XlaLegalizeTargetsTest() : context_(GetDefaultDialectRegistry()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op_test.cc
TEST(TfQuantOpTest, applyUniformQuantization) { MLIRContext context; OwningOpRef<ModuleOp> module(ModuleOp::create(UnknownLoc::get(&context))); OpBuilder builder(&module->getBodyRegion()); context.loadDialect<TF::TensorFlowDialect, quant::QuantizationDialect, func::FuncDialect>(); EmptyPatternRewriter pattern_rewriter(builder); Value value = CreateConstValue<float>(builder, module->getLoc(), {1024, 2},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 3K bytes - Viewed (0)