- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for registry_ (0.23 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)