- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 75 for registry_ (0.14 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/sparsecore/embedding_pipelining.cc
} } return false; } struct EmbeddingPipeliningPass : public ::impl::EmbeddingPipeliningPassBase<EmbeddingPipeliningPass> { void getDependentDialects(mlir::DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; }; bool UseEmbeddingPipelining(ModuleOp& module) { // Enable automated pipelining pass unless:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc
*registry, InferSparseSegmentOpCost<mlir::TF::SparseSegmentMeanOp>); RegisterCostFunction<mlir::TF::SparseSegmentSqrtNOp>( *registry, InferSparseSegmentOpCost<mlir::TF::SparseSegmentSqrtNOp>); RegisterCostFunction<mlir::TF::LookupTableFindV2Op>( *registry, InferLookupTableFindV2Cost); return registry; }(); return *registry; } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir.cc
TF_Status* status) { RegisterPasses(); mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::stablehlo::registerAllDialects(registry); registry.insert<mlir::shape::ShapeDialect>(); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> module; { mlir::StatusScopedDiagnosticHandler diagnostic_handler(&context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 19.3K bytes - Viewed (0) -
RELEASE.md
### TPU Enhancements * Adds 3D mesh support in TPU configurations ops. * Added TPU code for `FTRL` with `multiply_linear_by_lr`. * Silently adds a new file system registry at `gstpu`. * Support `restartType` in cloud tpu client. * Depend on a specific version of google-api-python-client. * Fixes apiclient import. ### Tracing and Debugging
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K 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) -
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/quantization/tensorflow/passes/prepare_quantize_drq.cc
// quantization case. class PrepareQuantizeDRQPass : public PassWrapper<PrepareQuantizeDRQPass, OperationPass<ModuleOp>> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, ::mlir::quant::QuantizationDialect, ::mlir::quantfork::QuantizationForkDialect>(); } public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0)