- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 32 for registry_ (0.11 sec)
-
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
<< ", FallbackEnabled: " << num_passes_fallback_enabled << ", Total: " << registry_->passes().size(); } GraphDebugInfo debug_info; mlir::DialectRegistry registry; RegisterDialects(registry); mlir::MLIRContext context(registry); GraphImportConfig import_config; import_config.graph_as_function = true; import_config.control_outputs = *control_ret_node_names;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 18.5K bytes - Viewed (0) -
tensorflow/cc/framework/gradients.cc
}; SymbolicGradientBuilder::SymbolicGradientBuilder( const Scope& scope, const ops::GradOpRegistry* registry, const std::vector<Output>& outputs, const std::vector<Output>& inputs, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) : scope_(scope), registry_(registry), outputs_(outputs), inputs_(inputs), grad_inputs_(grad_inputs),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
XlaCallModuleDeserializationPass> { public: void getDependentDialects(mlir::DialectRegistry ®istry) const override { XlaCallModuleDeserializationPassBase::getDependentDialects(registry); mlir::func::registerAllExtensions(registry); } void runOnOperation() override { ModuleOp module = getOperation(); SymbolTableCollection symbol_tables;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
std::iota(input_mapping->begin(), input_mapping->end(), 0); } static void RegisterDialects(mlir::DialectRegistry& registry) { mlir::RegisterAllTensorFlowDialects(registry); mlir::mhlo::registerAllMhloDialects(registry); mlir::stablehlo::registerAllDialects(registry); } // Checks if functions can be inlined after TF -> HLO legalization. Currently
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
} return success(); } }; class LiftTfliteFlexOpsPass : public impl::LiftTfliteFlexOpsPassBase<LiftTfliteFlexOpsPass> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LiftTfliteFlexOpsPass) void runOnOperation() override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc
StringRef getDescription() const final { return "Unfuses MHLO batch norm inference op into arithmetic ops"; } void getDependentDialects(::mlir::DialectRegistry ®istry) const override { // TODO(b/299514833): Remove TensorFlowDialect usage. registry.insert<shape::ShapeDialect, mlir::TF::TensorFlowDialect>(); } void runOnOperation() override { RewritePatternSet patterns(&getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.2K bytes - Viewed (0) -
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)