- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 138 for registry_ (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
class LegalizeTFGToTFE : public impl::LegalizeTFGToTFPassBase<LegalizeTFGToTFE> { void getDependentDialects(DialectRegistry ®istry) const override { RegisterAllTensorFlowDialects(registry); } void runOnOperation() override; }; } // namespace void LegalizeTFGToTFE::runOnOperation() { MLIRContext &context = getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc
"allowlist) as select Tensorflow ops"), llvm::cl::init(false)); // The following approach allows injecting opdefs in addition // to those that are already part of the global TF registry to be linked in // prior to importing the graph. The primary goal is for support of custom ops. // This is not intended to be a general solution for custom ops for the future
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils_test.cc
void Register(const std::string& op_name, OpRegistry* registry) { registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { op_reg_data->op_def.set_name(op_name); return absl::OkStatus(); }); } } // namespace TEST(TfTextUtilsTest, TestTfTextRegistered) { std::unique_ptr<OpRegistry> registry(new OpRegistry); Register("WhitespaceTokenizeWithOffsets", registry.get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:14:37 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/dialect_registration.h
if (include_extensions) { mlir::func::registerAllExtensions(registry); } } // Inserts all the TensorFlow dialects in the provided registry. This is // intended for tools that need to register dialects before parsing .mlir files. inline void RegisterAllTensorFlowDialects(DialectRegistry ®istry) { RegisterAllTensorFlowDialectsImpl(registry, true); } } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 15 16:15:16 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ir/tfrt_fallback_util_test.cc
"tensorflow/compiler/mlir/tfrt/tests/ir/testdata/test.mlir"); mlir::DialectRegistry registry; 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: Tue Jun 06 03:08:33 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_translate.cc
MlirToFlatBufferFileTranslateFunction, [](DialectRegistry& registry) { registry.insert<quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); mlir::RegisterAllTensorFlowDialects(registry); registry.insert<TFL::TensorFlowLiteDialect>(); registry.insert<arith::ArithDialect>(); registry.insert<func::FuncDialect>(); registry.insert<mlir::vhlo::VhloDialect>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/python/tfr_wrapper.cc
PYBIND11_MODULE(tfr_wrapper, m) { m.def("verify", [](std::string input) { mlir::DialectRegistry registry; registry.insert<mlir::arith::ArithDialect, mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, mlir::func::FuncDialect, mlir::shape::ShapeDialect, mlir::TFR::TFRDialect>(); mlir::MLIRContext ctx(registry); ctx.loadAllAvailableDialects(); llvm::SourceMgr source_mgr = llvm::SourceMgr();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 13 06:54:12 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_mlir_util_test.cc
mlir::DialectRegistry registry; mlir::registerAllDialects(registry); mlir::RegisterAllTensorFlowDialects(registry); mlir::registerAllPasses(); mlir::registerTensorFlowPasses(); EXPECT_TRUE(mlir::MlirOptMain(output_stream->os(), std::move(input_file), registry, mlir::MlirOptMainConfig{}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 18 13:40:21 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/not_differentiable.cc
for (int i = 0; i < grad_inputs.size(); i++) { grad_inputs[i] = nullptr; } return absl::OkStatus(); } Status RegisterNotDifferentiable(GradientRegistry* registry, const string& op) { return registry->Register(op, [](const ForwardOperation& op) { return new NotDifferentiableGradientFunction; }); } } // namespace gradients
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf_mlir_opt_main.cc
tensorflow::tfrt_compiler:: RegisterNonTPULowerClusterToRuntimeOpsPassPipeline(); mlir::DialectRegistry registry; mlir::RegisterCommonToolingDialects(registry); return failed( mlir::MlirOptMain(argc, argv, "TensorFlow pass driver\n", registry));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 4K bytes - Viewed (0)