- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 112 for registry_ (0.15 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/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) -
tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc
return absl::OkStatus(); } tensorflow::Status RunConverter(const PassPipelineCLParser& pass_pipeline) { DialectRegistry registry; registerAllDialects(registry); RegisterAllTensorFlowDialects(registry); mhlo::registerAllMhloDialects(registry); stablehlo::registerAllDialects(registry); registry.insert<mlir::func::FuncDialect, mlir::tf_type::TFTypeDialect, mlir::quant::QuantizationDialect>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
} static TranslateFromMLIRRegistration mlir_to_graph_translate( /*name=*/"mlir-to-graph", /*description=*/"convert mlir to graph", MlirToGraphTranslateFunction, [](DialectRegistry& registry) { mlir::RegisterAllTensorFlowDialects(registry); }); static LogicalResult MlirToGraphdefTranslateFunction( ModuleOp module, llvm::raw_ostream& output) { if (!module) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K bytes - Viewed (0)