Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 35 for unwrap (0.2 sec)

  1. tensorflow/c/eager/c_api_unified_experimental.cc

    void TF_DeleteOutputList(TF_OutputList* o) { delete unwrap(o); }
    void TF_OutputListSetNumOutputs(TF_OutputList* o, int num_outputs,
                                    TF_Status* s) {
      unwrap(o)->expected_num_outputs = num_outputs;
      unwrap(o)->outputs.clear();
      unwrap(o)->outputs.resize(num_outputs);
    }
    int TF_OutputListNumOutputs(TF_OutputList* o) {
      return unwrap(o)->outputs.size();
    }
    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/unified_api_testutil.h

          TF_NewStatus(), TF_DeleteStatus);
      TFE_Context* eager_ctx =
          TF_ExecutionContextGetTFEContext(wrap(ctx), status.get());
      TF_RETURN_IF_ERROR(StatusFromTF_Status(status.get()));
      TFE_TensorHandle* input_eager =
          TestTensorHandleWithDims<T, datatype>(eager_ctx, data, dims, num_dims);
      *tensor =
          unwrap(TF_CreateAbstractTensorFromEagerTensor(input_eager, status.get()));
      return absl::OkStatus();
    }
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_debug.cc

    extern "C" {
    
    TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo(
        TFE_TensorHandle* h, TF_Status* status) {
      tensorflow::TensorHandle* handle =
          TensorHandleFromInterface(tensorflow::unwrap(h));
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
      if (!status->status.ok()) {
        return nullptr;
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 11 01:20:50 GMT 2021
    - 2.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      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.
      if (step_id_.has_value()) {
        tensorflow::unwrap(op_.get())->SetStepId(step_id_.value());
      }
    
    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)
  5. tensorflow/c/eager/c_api_unified_experimental_eager.cc

                                                              TF_Status* s) {
      return wrap(static_cast<AbstractTensorHandle*>(unwrap(t)));
    }
    
    TFE_TensorHandle* TF_AbstractTensorGetEagerTensor(TF_AbstractTensor* at,
                                                      TF_Status* s) {
      auto handle = dyn_cast<ImmediateExecutionTensorHandle>(unwrap(at));
      if (!handle) {
        string msg =
    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)
  6. tensorflow/c/conversion_macros.h

      inline cpp_impl *unwrap(wrapper *w) {                                        \
        return reinterpret_cast<cpp_impl *>(w);                                    \
      }                                                                            \
                                                                                   \
      inline const cpp_impl *unwrap(const wrapper *w) {                            \
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon May 04 16:24:03 GMT 2020
    - 1.7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device.cc

          } else {
            TF_SetStatus(
                status, TF_INVALID_ARGUMENT,
                absl::StrCat(
                    "Got a non-parallel tensor ",
                    tensorflow::unwrap(absl::get<TFE_TensorHandle*>(input))
                        ->DebugString(),
                    " as input to a parallel operation. First pack non-parallel "
                    "tensors for each device into a parallel tensor explicitly.")
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_test.cc

        ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    
        auto cpu_arg =
            tensorflow::TensorHandleFromInterface(tensorflow::unwrap(hcpu));
        auto gpu_arg =
            tensorflow::TensorHandleFromInterface(tensorflow::unwrap(hgpu));
        auto gpu_device = gpu_arg->device();
        ASSERT_FALSE(cpu_arg->HasLocalMirror(gpu_device));
    
        TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  9. tensorflow/c/eager/c_api.cc

      tensorflow::unwrap(ctx)->StartStep();
    }
    
    void TFE_ContextEndStep(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->EndStep();
    }
    
    const TFE_OpAttrs* TFE_OpGetAttrs(const TFE_Op* op) {
      return tensorflow::wrap(tensorflow::unwrap(op)->GetOpAttrs());
    }
    
    void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs) {
      tensorflow::unwrap(op)->AddAttrs(tensorflow::unwrap(attrs));
    }
    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)
  10. tensorflow/c/eager/c_api_experimental.cc

      tensorflow::unwrap(ctx)->SetExecutorForThread(executor->executor());
    }
    
    TFE_Executor* TFE_ContextGetExecutorForThread(TFE_Context* ctx) {
      return new TFE_Executor(&tensorflow::unwrap(ctx)->Executor());
    }
    
    void TFE_HostAddressSpace(TFE_Context* ctx, TF_Buffer* buf) {
      auto address_space = tensorflow::DeviceNameUtils::AddressSpace(
          tensorflow::unwrap(ctx)->HostCPUParsedName());
    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)
Back to top