Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for Wrap (0.15 sec)

  1. tensorflow/c/eager/c_api_unified_experimental.cc

          s, tracing_ctx->AddParameter(static_cast<DataType>(dtype), partial_shape,
                                       &t));
      return wrap(t);
    }
    
    void TF_DeleteExecutionContext(TF_ExecutionContext* c) { unwrap(c)->Release(); }
    
    TF_AbstractOp* TF_NewAbstractOp(TF_ExecutionContext* c) {
      return wrap((unwrap(c)->CreateOperation()));
    }
    
    void TF_DeleteAbstractOp(TF_AbstractOp* op) { unwrap(op)->Release(); }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental_eager.cc

    using tensorflow::string;
    using tensorflow::unwrap;
    using tensorflow::wrap;
    using tensorflow::strings::StrCat;
    
    TF_ExecutionContext* TF_NewEagerExecutionContext(TFE_ContextOptions* options,
                                                     TF_Status* s) {
      TFE_Context* c_ctx = TFE_NewContext(options, s);
      if (TF_GetCode(s) != TF_OK) {
        return nullptr;
      }
      return wrap(static_cast<AbstractContext*>(unwrap(c_ctx)));
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jun 25 04:40:46 GMT 2020
    - 3.2K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_experimental.cc

      tensorflow::Tensor tensor(dtype, tensorflow::TensorShape({}));
      std::memcpy(tensorflow::TensorCApi::Buffer(tensor)->data(), data, len);
    
      status->status = absl::OkStatus();
      return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor));
    }
    
    // Set server_def on the context, possibly updating it.
    TF_CAPI_EXPORT extern void TFE_EnableCollectiveOps(TFE_Context* ctx,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.cc

    #if !defined(IS_MOBILE_PLATFORM)
      eager_context->SetDistributedManager(
          std::make_unique<tensorflow::EagerContextDistributedManager>(
              eager_context));
    #endif  // !IS_MOBILE_PLATFORM
      return tensorflow::wrap(eager_context);
    }
    
    void TFE_DeleteContext(TFE_Context* ctx) {
      if (ctx == nullptr) {
        return;
      }
    
      // ctx->RefCountIsOne() should be true here.
      tensorflow::unwrap(ctx)->Release();
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  5. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

        if (TF_GetCode(status) != TF_OK) return;
      }
      std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs);
      int real_num_outputs = expected_max_outputs;
      TFE_OpSetCancellationManager(op_.get(), wrap(cancellation_manager_), status);
      if (TF_GetCode(status) != TF_OK) return;
    
      // unwrap op_ and set step_id only if valid step id value was set.
      // Currently only required for non-TFRT use cases, e.g., EagerOp.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  6. tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc

                                                absl::string_view result,
                                                size_t read) {
      // Result isn't a null-terminated string so we have to wrap it inside a
      // `string_view`
      if (length == read && content_view.substr(offset, length) ==
                                absl::string_view(result).substr(0, read))
        return ::testing::AssertionSuccess();
      else
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Aug 31 12:04:23 GMT 2020
    - 24.9K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_experimental.cc

    }
    
    void TFE_ContextOptionsSetTfrt(TFE_ContextOptions* options, bool use_tfrt) {
      options->use_tfrt = use_tfrt;
    }
    
    TFE_CancellationManager* TFE_NewCancellationManager() {
      return tensorflow::wrap(new tensorflow::CancellationManager);
    }
    
    void TFE_CancellationManagerStartCancel(
        TFE_CancellationManager* cancellation_manager) {
      tensorflow::unwrap(cancellation_manager)->StartCancel();
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  8. tensorflow/c/eager/unified_api_testutil.cc

      std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
          TF_NewStatus(), TF_DeleteStatus);
      TFE_TensorHandle* result_t =
          TF_AbstractTensorGetEagerTensor(wrap(t), status.get());
      TF_RETURN_IF_ERROR(StatusFromTF_Status(status.get()));
      *result_tensor = TFE_TensorHandleResolve(result_t, status.get());
      return StatusFromTF_Status(status.get());
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 5.7K bytes
    - Viewed (0)
Back to top