- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 99 for registry_ (0.43 sec)
-
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) -
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) -
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/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) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_algorithm.py
_CalibrationMethod = ( stablehlo_quant_config_pb2.CalibrationOptions.CalibrationMethod ) _REGISTRY = {} def _implements(calib_method: _CalibrationMethod): def decorator(cls): assert calib_method not in _REGISTRY _REGISTRY[calib_method] = cls return cls return decorator class _CalibrationAlgorithmBase(abc.ABC):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 14.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)