- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 138 for registry_ (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/shape_inference_utils.h
#include "tensorflow/core/ir/utils/shape_inference_utils.h" namespace mlir { class Operation; namespace TF { // Runs TensorFlow shape inference associated to the op type registered in the // TensorFlow op registry based on the Graph version, operands, and attributes. // Invoking this shape function will create conversions of parameters to the // TensorFlow Graph equivalent data structures and back to MLIR equivalent data
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 1.9K bytes - Viewed (0) -
.github/workflows/sigbuild-docker-branch.yml
password: ${{ secrets.DOCKERHUB_TOKEN }} - name: Login to GCR uses: docker/login-action@f4ef78c080cd8ba55a85445d5b36e214a81df20a # v2.1.0 with: registry: gcr.io username: _json_key password: ${{ secrets.GCP_CREDS }} - name: Generate variables for cache busting and tag naming run: |
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 18:43:43 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/register_common_dialects.h
#include "mlir/IR/DialectRegistry.h" // from @llvm-project namespace mlir { // Inserts common Tensorflow dialects used for offline tools. void RegisterCommonToolingDialects(mlir::DialectRegistry& registry); }; // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 21 20:13:57 UTC 2023 - 1.1K 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/lite/quantization/lite/quantize_weights.cc
absl::flat_hash_set<std::string> denylisted_mlir_op_names; for (auto& entry : denylisted_ops) { denylisted_mlir_op_names.insert(TfLiteToMlir(entry)); } DialectRegistry registry; MLIRContext context(registry); StatusScopedDiagnosticHandler statusHandler(&context, /*propagate=*/true); // Import input_model to a MLIR module
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
class LowerGlobalsToMlProgram : public impl::LowerGlobalsToMlProgramPassBase<LowerGlobalsToMlProgram> { public: void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<mlir::tf_saved_model::TensorFlowSavedModelDialect, ml_program::MLProgramDialect>(); } void runOnOperation() override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K 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)