- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 39 for registry_ (0.38 sec)
-
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
using tensorflow::tracing::TracingOperation; using tensorflow::tracing::TracingTensorHandle; namespace { void RegisterDialects(mlir::MLIRContext& ctx) { mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); ctx.appendDialectRegistry(registry); ctx.loadAllAvailableDialects(); } Status ConvertDataTypeToTensor(tensorflow::DataType dtype, Builder builder, Type* type) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
}; class PrepareCompositeFunctionsPass : public impl::PrepareCompositeFunctionsPassBase< PrepareCompositeFunctionsPass> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TFL::TensorFlowLiteDialect>(); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(PrepareCompositeFunctionsPass) explicit PrepareCompositeFunctionsPass() {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
// training quantization simpler. class PrepareQuantizePass : public PassWrapper<PrepareQuantizePass, OperationPass<func::FuncOp>> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, ::mlir::quant::QuantizationDialect, ::mlir::quantfork::QuantizationForkDialect>(); } public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir.cc
TF_Status* status) { RegisterPasses(); mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::stablehlo::registerAllDialects(registry); registry.insert<mlir::shape::ShapeDialect>(); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> module; { mlir::StatusScopedDiagnosticHandler diagnostic_handler(&context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_algorithm.py
_CalibrationMethod = ( stablehlo_quant_config_pb2.CalibrationOptions.CalibrationMethod ) _REGISTRY = {} def _implements(calib_method: _CalibrationMethod): def decorator(cls): assert calib_method not in _REGISTRY _REGISTRY[calib_method] = cls return cls return decorator class _CalibrationAlgorithmBase(abc.ABC):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 14.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize_drq.cc
// quantization case. class PrepareQuantizeDRQPass : public PassWrapper<PrepareQuantizeDRQPass, OperationPass<ModuleOp>> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, ::mlir::quant::QuantizationDialect, ::mlir::quantfork::QuantizationForkDialect>(); } public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
// This is a brief description of the pass. return "Insert custom aggregation ops for the calibration procedure"; } void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; private: enum TestCase { TEST_CASE_MIN_MAX, TEST_CASE_AVERAGE_MIN_MAX,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
// This is a brief description of the pass. return "Apply graph optimizations such as fusing and constant folding to " "prepare lifting."; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, arith::ArithDialect>(); } void runOnOperation() override; private: Option<OpSet> op_set_{ *this, "target-opset", llvm::cl::init(OpSet::TF),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
struct DecomposeReduceDatasetPass : public impl::DecomposeReduceDatasetPassBase<DecomposeReduceDatasetPass> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<tf_device::TensorFlowDeviceDialect>(); } void runOnOperation() override; }; // Create the AnonymousIterator for `reduce_dataset` with `dataset_types` using // `builder`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
: public PassWrapper<RaiseToTFOpsPass, OperationPass<func::FuncOp>> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(RaiseToTFOpsPass) void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TFRDialect, TF::TensorFlowDialect, scf::SCFDialect, arith::ArithDialect, func::FuncDialect>(); } explicit RaiseToTFOpsPass(std::optional<ModuleOp> tfr_module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0)