- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 91 for registry_ (0.12 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_custom_aggregation_op_to_quant_stats.cc
// This is a brief description of the pass. return "Convert tf.CustomAggregator op to quant.Stats"; } void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TF::TensorFlowDialect>(); registry.insert<quant::QuantizationDialect>(); registry.insert<quantfork::QuantizationForkDialect>(); } void runOnOperation() override; }; class ConvertCustomAggregationOpToQuantStats
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc
std::vector<TensorShape>& arg_shapes, tpu::TPUCompileMetadataProto& metadata_proto) { mlir::DialectRegistry registry; mlir::RegisterCommonToolingDialects(registry); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; TF_RETURN_IF_ERROR( DeserializeMlirModule(mlir_module_str, &context, &mlir_module));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0) -
.github/workflows/sigbuild-docker-presubmit.yml
tags: | gcr.io/tensorflow-sigs/build:${{ github.event.number }}-${{ matrix.python-version }} cache-from: | type=registry,ref=tensorflow/build:latest-${{ matrix.python-version }} type=registry,ref=gcr.io/tensorflow-sigs/build:${{ github.event.number }}-${{ matrix.python-version }} cache-to: type=inline - name: Add a comment with the pushed containers
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 18:43:43 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/tf_to_stablehlo_translate.cc
llvm::cl::ParseCommandLineOptions(argc, argv, "TF Saved Model to Stablehlo converter\n"); mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); context.loadAllAvailableDialects(); bool is_input_mlir_module = absl::EndsWith(input_path, ".mlir"); std::vector<std::string> exported_model_signatures_in_vector =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst_test.cc
using tsl::StatusOr; // MockSuccessPass reports errors but doesn't fail. class MockSuccessPass : public PassWrapper<MockSuccessPass, OperationPass<ModuleOp>> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MockSuccessPass) explicit MockSuccessPass() = default; private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
StringRef getDescription() const final { return "Smuggle disallowed ops via stablehlo.custom_calls"; } void getDependentDialects(DialectRegistry& registry) const final { registry.insert<mlir::stablehlo::StablehloDialect>(); } void runOnOperation() override { RewritePatternSet patterns(&getContext()); patterns.add<SmuggleOpPattern<TF::ResizeBilinearOp>>(&getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_gpu.cc
return "tfl-device-transform-gpu"; } llvm::StringRef getDescription() const final { return "Suitable transformation for gpu only."; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; }; void DeviceTransformGPUPass::runOnOperation() { auto func = getOperation(); auto* ctx = &getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 07 18:43:51 UTC 2022 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
StringPiece tfr_raw_text, mlir::MLIRContext* mlir_ctx) { mlir_ctx->allowUnregisteredDialects(/*allow=*/true); // Load dialects involved in the conversion mlir::DialectRegistry registry; // clang-format off registry.insert<mlir::arith::ArithDialect, mlir::func::FuncDialect, mlir::scf::SCFDialect, mlir::shape::ShapeDialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
.github/workflows/sigbuild-docker.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: Grab the upcoming TF version to tag this container run: |
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 18:43:43 UTC 2023 - 3.8K bytes - Viewed (0)