Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for RegisterFunction (0.25 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to top