- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 42 for registry_ (0.14 sec)
-
tensorflow/c/experimental/gradients/math_grad_test.cc
} // TODO(srbs): Rename ops::Add to ops::AddV2 and AddRegister to // AddV2Registerer. status_ = registry_.Register("AddV2", AddRegisterer); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); ASSERT_NO_FATAL_FAILURE(CompareNumericalAndAutodiffGradients( AddModel, BuildGradModel(AddModel, registry_), immediate_execution_ctx_.get(), {x.get(), y.get()}, UseFunction())); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 17:32:14 UTC 2023 - 16.3K bytes - Viewed (0) -
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/c/eager/gradients.cc
Status GradientRegistry::Register( const string& op_name, GradientFunctionFactory gradient_function_factory) { auto iter = registry_.find(op_name); if (iter != registry_.end()) { const string error_msg = "Gradient already exists for op: " + op_name + "."; return errors::AlreadyExists(error_msg); } registry_.insert({op_name, gradient_function_factory}); return absl::OkStatus(); } Status GradientRegistry::Lookup(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K 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/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
return "This pass will legalize TFLite custom Ops to StableHLO ops."; } private: void runOnOperation() override; void getDependentDialects(DialectRegistry& registry) const override { mlir::stablehlo::registerAllDialects(registry); } inline TFL::ConstBytesAttr CustomOption(OpBuilder* builder, const std::string& content) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K 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/quantization/import_quant_stats_pass.cc
// This is a brief description of the pass. return "Import quantization stats to the model"; } void runOnOperation() override; void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); } // Parses the serialized quant stats protobuf and initialize the internal
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 9.9K 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)