- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 61 for registry_ (0.16 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/context.h
// Creates an MLIRContext with the extensions required for quantization are // registered. inline std::unique_ptr<MLIRContext> CreateMlirContextForQuantization() { DialectRegistry registry{}; func::registerAllExtensions(registry); return std::make_unique<MLIRContext>(registry); } } // namespace mlir::quant::stablehlo
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 11 16:08:42 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
std::vector<AbstractTensorHandle*> outputs(1); GradientRegistry registry; s = RegisterGradients(®istry); ASSERT_EQ(errors::OK, s.code()) << s.message(); auto tape = std::make_unique<Tape>(/*persistent=*/false); s = Execute(check_numerics_op.get(), ctx.get(), absl::MakeSpan(outputs), &num_retvals, &forward_op, tape.get(), registry); ASSERT_EQ(errors::OK, s.code()) << s.message(); string read_message;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/debug/debug_test.cc
mlir::registerPassManagerCLOptions(); mlir::DialectRegistry registry; registry.insert<mlir::BuiltinDialect>(); registry.insert<mlir::arith::ArithDialect>(); registry.insert<mlir::func::FuncDialect>(); registry.insert<mlir::TFL::TensorFlowLiteDialect>(); return registry; }()) { context_.loadAllAvailableDialects();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 11:15:16 UTC 2024 - 9.7K 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/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/tfrt/tests/saved_model/saved_model_test.cc
std::string saved_model_mlir_path = tensorflow::GetDataDependencyFilepath( "tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/test.mlir"); mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); auto module = mlir::parseSourceFile<mlir::ModuleOp>(saved_model_mlir_path, &context); ASSERT_TRUE(module); std::vector<std::string> inputs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 13 01:17:29 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/analysis/update_op_cost_in_tfrt_mlir_test.cc
"tensorflow/compiler/mlir/tfrt/tests/analysis/testdata/test.mlir"); mlir::DialectRegistry registry; tfrt::RegisterTFRTDialects(registry); registry.insert<tfrt::fallback_async::FallbackAsyncDialect>(); registry.insert<tfrt::fallback_sync::FallbackSyncDialect>(); mlir::MLIRContext context(registry); auto module = mlir::parseSourceFile<mlir::ModuleOp>(saved_model_mlir_path, &context); ASSERT_TRUE(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 22:52:12 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/pywrap_tensorflow_to_stablehlo_lib.cc
const std::vector<std::string>& exported_model_signatures, const std::vector<std::string>& tag_names, absl::string_view input_arg_shapes_str) { mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); context.loadAllAvailableDialects(); auto module = TfToStablehlo(input_path, &context, exported_model_signatures, tag_names,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
} } TF_DeleteTensor(analytical_tensor); delete[] danalytical; } Model BuildGradModel(Model forward, GradientRegistry registry) { return [forward_model = std::move(forward), grad_registry = std::move(registry)]( AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) -> Status {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 5K bytes - Viewed (0)