- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for registry_ (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
let description = [{ Fixed point shape refinement pass that utilizes the shape functions registered on ops using the InferTypeOpInterface as well as by bridging to the TensorFlow op registry's shape functions. This is an interprocedural pass that propagates information across function calls/control flow operations where possible (the GuaranteeAllFuncsOneUsePass is often run
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
} void LoadImporterDialects(mlir::MLIRContext& context) { // Load dialects involved in the conversion mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialectsImpl(registry, false); context.appendDialectRegistry(registry); for (llvm::StringRef name : registry.getDialectNames()) context.getOrLoadDialect(name); } absl::StatusOr<std::string> GetDenseTensorNameFromTensorInfo(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// It is a terrible idea to have local MLIR contexts so we need to // register extensions here, again. mlir::DialectRegistry registry; registry.insert<mlir::func::FuncDialect>(); mlir::func::registerAllExtensions(registry); xla_call_module_context_.appendDialectRegistry(registry); auto l = tensorflow::XlaCallModuleLoader::Create( &xla_call_module_context_, op.getVersion(), op.getModule().str(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
} func.return // expected-remark@above {{ID: 25}} // expected-remark@above {{Sinks: {24}}} } // ----- // Tests that the pass tracks control dependencies based on TF op registry // statefulness flag, for ops not yet defined in ODS. // CHECK-LABEL: func @tf_registry_ops func.func @tf_registry_ops( // expected-remark@above {{ID: 8}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0)