- Sort Score
- Result 10 results
- Languages All
Results 41 - 45 of 45 for GetDependentDialects (0.24 sec)
-
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
class RaiseToTFOpsPass : public PassWrapper<RaiseToTFOpsPass, OperationPass<func::FuncOp>> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(RaiseToTFOpsPass) void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TFRDialect, TF::TensorFlowDialect, scf::SCFDialect, arith::ArithDialect, func::FuncDialect>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
: public impl::TPUClusterFormationPassBase<TPUClusterFormationPass> { public: explicit TPUClusterFormationPass(bool strict_clusters) : strict_clusters_(strict_clusters) {} void getDependentDialects(DialectRegistry& registry) const override { registry.insert<mlir::tf_device::TensorFlowDeviceDialect>(); } void runOnOperation() override; private: bool strict_clusters_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
"_tpu_compilation_status"; namespace mlir { namespace TFDevice { namespace { struct EmbeddingSequencingPass : public ::impl::EmbeddingSequencingPassBase<EmbeddingSequencingPass> { void getDependentDialects(mlir::DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; }; template <typename InputContainer>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
} StringRef getDescription() const final { // 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:
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/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
return true; } } 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) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0)