- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 75 for registry_ (0.21 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
// This is a brief description of the pass. return "Insert custom aggregation ops for the calibration procedure"; } void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; private: enum TestCase { TEST_CASE_MIN_MAX, TEST_CASE_AVERAGE_MIN_MAX,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
// This is a brief description of the pass. return "Apply graph optimizations such as fusing and constant folding to " "prepare lifting."; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, arith::ArithDialect>(); } void runOnOperation() override; private: Option<OpSet> op_set_{ *this, "target-opset", llvm::cl::init(OpSet::TF),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
struct DecomposeReduceDatasetPass : public impl::DecomposeReduceDatasetPassBase<DecomposeReduceDatasetPass> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<tf_device::TensorFlowDeviceDialect>(); } void runOnOperation() override; }; // Create the AnonymousIterator for `reduce_dataset` with `dataset_types` using // `builder`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
: 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>(); } explicit RaiseToTFOpsPass(std::optional<ModuleOp> tfr_module,
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/lite/stablehlo/transforms/fold_broadcast_pass.cc
StringRef getDescription() const final { return "Constant folds BroadcastInDimOp before binary elementwise ops"; } void getDependentDialects(::mlir::DialectRegistry ®istry) const override {} void runOnOperation() override { RewritePatternSet patterns(&getContext()); patterns.add<FoldBroadcastInDimBeforeMulOp>(&getContext()); patterns.add(ConstantFoldMul);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K 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/tensorflow/transforms/sparsecore/embedding_sequencing.cc
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> std::vector<Type> GetValueTypes(const InputContainer& input) {
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/tensorflow_to_stablehlo/python/pywrap_tensorflow_to_stablehlo_lib.cc
const std::vector<std::string>& exported_model_signatures, const std::vector<std::string>& tag_names, absl::string_view input_arg_shapes_str) { mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); context.loadAllAvailableDialects(); auto module = TfToStablehlo(input_path, &context, exported_model_signatures, tag_names,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
static void RegisterMlirInputDialects(mlir::DialectRegistry& registry) { // TODO(b/259459405): Remove support for stablehlo as an input. registry .insert<mlir::arith::ArithDialect, mlir::func::FuncDialect, mlir::TF::TensorFlowDialect, mlir::stablehlo::StablehloDialect, mlir::quant::QuantizationDialect>(); mlir::func::registerAllExtensions(registry); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K 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)