- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 145 for registry_ (0.24 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_quantized_functions.cc
StringRef getDescription() const final { // This is a brief description of the pass. return "Insert quantized functions into the module"; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, func::FuncDialect>(); } private: void runOnOperation() override; // Returns the function library for the given quantization method and opset // pair.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h
// name should be unique to the graph it is being inserted to. If the // `ignore_unregistered_attrs` argument is set to true, the attributes which are // not in the op registry will be ignored. If the `ignore_unregistered_attrs` // argument is not set to true, _output_shapes attribute is added to nodes with // ShapedType for the leading values with ShapedType in the results of the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
// Explicitly disable dumping Op details on failures. module.getContext()->printOpOnDiagnostic(false); mlir::DialectRegistry registry; mlir::func::registerAllExtensions(registry); module.getContext()->appendDialectRegistry(registry); mlir::StatusScopedDiagnosticHandler status_handler(module.getContext(), /*propagate=*/true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/converter_python_api.cc
PyExc_ValueError, "Failed to parse opdefs at index %d of custom op defs argument: %s", i, tf_opdefs); return nullptr; } // Register extra opdefs to TensorFlow global op registry. tensorflow::OpRegistry::Global()->Register( [opdef]( tensorflow::OpRegistrationData* op_reg_data) -> tensorflow::Status { *op_reg_data = tensorflow::OpRegistrationData(opdef);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
public: explicit TPUClusterFormationPass(bool strict_clusters) : strict_clusters_(strict_clusters) {} void getDependentDialects(DialectRegistry& registry) const override { registry.insert<mlir::tf_device::TensorFlowDeviceDialect>(); } void runOnOperation() override; private: bool strict_clusters_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
// This is a brief description of the pass. return "Replace quantization candidates with composite functions into the " "module"; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; private: QuantizationOptions quant_options_; bool test_mode_; Option<OpSet> op_set_{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
// This is a brief description of the pass. return "Quantize composite functions with QDQ input/outputs."; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); } private: void runOnOperation() override;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
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)