- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for registry_ (0.14 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
// This is a brief description of the pass. return "Quantize composite functions with QDQ input/outputs."; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); } private: void runOnOperation() override;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
// // Arg --> B --> C --> D --> Retval class Subgraph { public: // Creates a graph to build the subgraph in, if it doesn't already exist, // using the same op registry and versions as graph_in. Node* MakeNodeImage(const Graph* graph_in, Node* node); // Returns the graph the subgraph is being built in. Graph* GetGraph() const;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K 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/lite/flatbuffer_import.cc
const std::vector<std::string>& ordered_output_arrays, bool experimental_prune_unreachable_nodes_unconditionally, const bool disable_vhlo_to_stablehlo) { mlir::DialectRegistry registry; registry.insert<mlir::arith::ArithDialect, mlir::func::FuncDialect, mlir::quant::QuantizationDialect, mlir::quantfork::QuantizationForkDialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
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 Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0)