- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for RegisterFunction (0.25 sec)
-
tensorflow/cc/experimental/libtf/function.cc
{ const tensorflow::FunctionDef* fdef = nullptr; TF_RETURN_IF_ERROR(trace->GetFunctionDef(&fdef)); fname = fdef->signature().name(); } // TODO(srbs): Update RegisterFunction to accept AbstractFunctionPtr. TF_RETURN_IF_ERROR(ctx->RegisterFunction(trace.get())); auto cleanup = absl::MakeCleanup( [fname, ctx]() { ctx->RemoveFunction(fname).IgnoreError(); }); auto call_op = AbstractOperationPtr(ctx->CreateOperation());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 04 19:49:06 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
->Finalize(&output_list, &func)); scoped_func.reset(func); for (auto output : output_list.outputs) { output->Unref(); } TF_RETURN_IF_ERROR(ctx->RegisterFunction(func)); } AbstractOperationPtr fn_op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr)); for (auto input : inputs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
TF_AbstractFunction* func, TF_Status* s) { tsl::Set_TF_Status_from_Status(s, unwrap(ctx)->RegisterFunction(unwrap(func)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
*f = new GraphFunction(std::move(func->record->fdef())); TF_DeleteFunction(func); TF_RETURN_IF_ERROR(StatusFromTF_Status(s)); TF_DeleteStatus(s); return absl::OkStatus(); } Status RegisterFunction(AbstractFunction* func) override { return errors::Unimplemented( "Registering graph functions has not been implemented yet."); } Status RemoveFunction(const string& func) override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
const tensorflow::PartialTensorShape& shape, TracingTensorHandle** handle) override; Status Finalize(OutputList* outputs, AbstractFunction** f) override; Status RegisterFunction(AbstractFunction* func) override { return Unimplemented( "Registering graph functions has not been implemented yet."); } Status RemoveFunction(const string& func) override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0)