- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 39 for registry_ (0.46 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
StringRef getDescription() const final { // This is a brief description of the pass. return "Apply converting Tensorflow Xla ops to non-xla ops."; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, arith::ArithDialect>(); } void runOnOperation() override; }; // Generate an einsum equation from the given DotDimensionNumber. std::string CreateEinsumEquation(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
class LegalizeTFGToTFE : public impl::LegalizeTFGToTFPassBase<LegalizeTFGToTFE> { void getDependentDialects(DialectRegistry ®istry) const override { RegisterAllTensorFlowDialects(registry); } void runOnOperation() override; }; } // namespace void LegalizeTFGToTFE::runOnOperation() { MLIRContext &context = getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc
return absl::OkStatus(); } tensorflow::Status RunConverter(const PassPipelineCLParser& pass_pipeline) { DialectRegistry registry; registerAllDialects(registry); RegisterAllTensorFlowDialects(registry); mhlo::registerAllMhloDialects(registry); stablehlo::registerAllDialects(registry); registry.insert<mlir::func::FuncDialect, mlir::tf_type::TFTypeDialect, mlir::quant::QuantizationDialect>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// shape assignment in shape_inference. std::unique_ptr<OperationPass<func::FuncOp>> CreateReplicateTensorListInitOpsPass(); // Performs Shape Inference on the TensorFlow dialect using the global registry. std::unique_ptr<OperationPass<ModuleOp>> CreateTFShapeInferencePass( ArrayRef<ArrayRef<int64_t>> input_shapes = {}); // Performs TF.data optimizations.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
// back to do it properly in the future mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); registry .insert<mlir::func::FuncDialect, mlir::stablehlo::StablehloDialect>(); context.appendDialectRegistry(registry); } absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> module; std::unordered_set<std::string> tags; tensorflow::GraphImportConfig specs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
"initializer functions will be removed after this pass."; } void runOnOperation() override; private: void getDependentDialects(DialectRegistry& registry) const override { registry .insert<TF::TensorFlowDialect, tf_executor::TensorFlowExecutorDialect, tf_saved_model::TensorFlowSavedModelDialect>(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
// Explicitly disable dumping Op details on failures. module.getContext()->printOpOnDiagnostic(false); mlir::DialectRegistry registry; mlir::func::registerAllExtensions(registry); module.getContext()->appendDialectRegistry(registry); mlir::StatusScopedDiagnosticHandler status_handler(module.getContext(), /*propagate=*/true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/converter_python_api.cc
PyExc_ValueError, "Failed to parse opdefs at index %d of custom op defs argument: %s", i, tf_opdefs); return nullptr; } // Register extra opdefs to TensorFlow global op registry. tensorflow::OpRegistry::Global()->Register( [opdef]( tensorflow::OpRegistrationData* op_reg_data) -> tensorflow::Status { *op_reg_data = tensorflow::OpRegistrationData(opdef);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
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/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
// This is a brief description of the pass. return "Replace quantization candidates with composite functions into the " "module"; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; private: QuantizationOptions quant_options_; bool test_mode_; Option<OpSet> op_set_{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0)