- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 99 for registry_ (0.13 sec)
-
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/c/experimental/gradients/tape/tape_operation.cc
namespace tensorflow { namespace gradients { TapeOperation::TapeOperation(AbstractOperation* parent_op, Tape* tape, const GradientRegistry& registry) : AbstractOperation(kTape), parent_op_(parent_op), tape_(tape), registry_(registry) { // TODO(b/172003047): Consider making AbstractOperation RefCounted. // parent_op_->Ref(); } void TapeOperation::Release() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 06:16:45 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tools/stablehlo_quant_opt.cc
mlir::tf_executor::TensorFlowExecutorDialect, mlir::vhlo::VhloDialect>(); mlir::mhlo::registerAllMhloDialects(registry); mlir::func::registerAllExtensions(registry); return failed( mlir::MlirOptMain(argc, argv, "StableHLO quant Pass Driver\n", registry));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 07:37:34 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc
"or compilation failure from the MLIR bridge (full fallback mode)."; mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::mhlo::registerAllMhloDialects(registry); mlir::stablehlo::registerAllDialects(registry); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; TF_RETURN_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
} private: void runOnOperation() override; void getDependentDialects(DialectRegistry ®istry) const override { mlir::mhlo::registerAllMhloDialects(registry); mlir::stablehlo::registerAllDialects(registry); registry.insert<mlir::func::FuncDialect, mlir::arith::ArithDialect>(); registry.insert<shape::ShapeDialect>(); } public: StringRef getArgument() const final { return "tf-mhlo"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K 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/quantization/tensorflow/tf_to_quant.cc
LegalizeTFToQuant(const LegalizeTFToQuant &) {} /// Performs the lowering to Quant ops dialect. void runOnOperation() override; void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); } StringRef getArgument() const final { // This is the argument used to refer to the pass in
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.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)