- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 112 for registry_ (0.28 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc
// This is a brief description of the pass. return "Add DumpTensor ops after quantizable ops"; } void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TF::TensorFlowDialect>(); registry.insert<quant::QuantizationDialect>(); registry.insert<quantfork::QuantizationForkDialect>(); } private: void runOnOperation() override; Option<DebuggerType> debugger_type_{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
} } TF_DeleteTensor(analytical_tensor); delete[] danalytical; } Model BuildGradModel(Model forward, GradientRegistry registry) { return [forward_model = std::move(forward), grad_registry = std::move(registry)]( AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) -> Status {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc
xla::CompileOnlyClient* client, XlaCompiler::CompilationResult* compilation_result) { mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::mhlo::registerAllMhloDialects(registry); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; TF_RETURN_IF_ERROR(DeserializeMlirModule(mlir_computation.mlir_module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_graph_optimization_pass.cc
class GraphOptPass : public mlir::PassWrapper<GraphOptPass, mlir::OperationPass<mlir::ModuleOp>> { void getDependentDialects(mlir::DialectRegistry& registry) const override { mlir::RegisterAllTensorFlowDialects(registry); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(GraphOptPass) explicit GraphOptPass(std::vector<tensorflow::GraphOptimizationPass*> passes)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_tf_xla_call_module_to_stablehlo_pass.cc
} StringRef getDescription() const final { return "Legalize TF_XlaCallModule Op to stablehlo"; } void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<stablehlo::StablehloDialect, vhlo::VhloDialect, quant::QuantizationDialect, shape::ShapeDialect>(); } void runOnOperation() override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 25 09:43:18 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc
// passes. Currently, per-channel quantization only supports 1D results. class PreprocessOpPass : public PassWrapper<PreprocessOpPass, OperationPass<ModuleOp>> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, QuantizationDialect, quantfork::QuantizationForkDialect>(); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(PreprocessOpPass)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
: public impl::ReduceWhileOperandsPassBase<ReduceWhileOperandsPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ReduceWhileOperandsPass) void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TFL::TensorFlowLiteDialect, TF::TensorFlowDialect>(); } void runOnOperation() override; }; LogicalResult FindImplicityProducers( const std::vector<uint64_t> &explicitly_consumed_ids,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
StringRef getDescription() const final { return "Fallback TF-Quantization-unsupported ops to TFLite Flex ops."; } void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TFL::TensorFlowLiteDialect>(); } private: // The mode of TF Quantization, might indicate different users/devices. Option<std::string> mode_{*this, "mode",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
"Lower threshold of the constant size for unfreezing. Constants " "smaller than this value will not be converted to variables.")); } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, tf_saved_model::TensorFlowSavedModelDialect>(); } // Lower-bound threshold for the size of the constant in bytes. Constants
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K 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)