Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for CTX (0.16 sec)

  1. tensorflow/c/eager/c_api_unified_experimental_test.cc

      TF_ExecutionContext* ctx = TF_NewEagerExecutionContext(opts, status.get());
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
      TFE_DeleteContextOptions(opts);
    
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
      // Build an abstract input tensor.
      TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 19 21:44:52 GMT 2023
    - 39.1K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_experimental_test.cc

      TFE_Executor* old_executor = TFE_ContextGetExecutorForThread(ctx);
      TFE_Executor* executor = TFE_NewExecutor(
          /*is_async=*/async, /*enable_streaming_enqueue=*/true,
          /*in_flight_nodes_limit=*/0);
      TFE_ContextSetExecutorForThread(ctx, executor);
    
      TFE_TensorHandle* m = TestMatrixTensorHandle(ctx);
      TFE_Op* matmul = MatMulOp(ctx, m, m);
      TFE_TensorHandle* retvals[2] = {nullptr, nullptr};
      int num_retvals = 2;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
  3. tensorflow/c/eager/dlpack.cc

        return nullptr;
      }
    
      TensorReference tensor_ref(*tensor);  // This will call buf_->Ref()
      auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref);
      tf_dlm_tensor_ctx->reference = tensor_ref;
    
      DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor;
      dlm_tensor->manager_ctx = tf_dlm_tensor_ctx;
      dlm_tensor->deleter = &DLManagedTensorDeleter;
      dlm_tensor->dl_tensor.device = tf_dlm_context;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern void TFE_DeleteContext(TFE_Context* ctx);
    TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
                                                                TF_Status* status);
    
    // Clears the internal caches in the TFE context. Useful when reseeding random
    // ops.
    TF_CAPI_EXPORT extern void TFE_ContextClearCaches(TFE_Context* ctx);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
  5. tensorflow/c/eager/c_api_experimental.h

    // this context.
    TF_CAPI_EXPORT extern void TFE_ContextEnableGraphCollection(TFE_Context* ctx);
    
    // Disables only graph collection in RunMetadata on the functions executed from
    // this context.
    TF_CAPI_EXPORT extern void TFE_ContextDisableGraphCollection(TFE_Context* ctx);
    
    // TODO(fishx): Move these monitoring APIs into a separate file.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients.cc

    AbstractTensorHandle* TapeTensor::ZerosLike() const { return nullptr; }
    
    class TapeVSpace
        : public eager::VSpace<AbstractTensorHandle, GradientFunction, TapeTensor> {
     public:
      explicit TapeVSpace(AbstractContext* ctx) : ctx_(ctx) {}
      ~TapeVSpace() override {}
    
      // Returns the number of elements in the gradient tensor.
      int64_t NumElements(AbstractTensorHandle* tensor) const override;
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_test_util.cc

    using tensorflow::tstring;
    
    TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, float value) {
      float data[] = {value};
      TF_Status* status = TF_NewStatus();
      TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, nullptr, 0, status);
      memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t));
      TFE_TensorHandle* th = TFE_NewTensorHandleFromTensor(ctx, t, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  8. tensorflow/c/experimental/gradients/math_grad.cc

    namespace gradients {
    namespace {
    
    static Status SafeConj(AbstractContext* ctx, AbstractTensorHandle* const input,
                           AbstractTensorHandle** output, const char* name) {
      auto dtype = input->DataType();
      if (DataTypeIsFloating(BaseType(dtype)) ||
          DataTypeIsInteger(BaseType(dtype))) {
        return tensorflow::ops::Identity(ctx, input, output, name);
      } else if (!DataTypeIsComplex(BaseType(dtype)) &&
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 15.2K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api.cc

    }
    
    void TFE_DeleteContext(TFE_Context* ctx) {
      if (ctx == nullptr) {
        return;
      }
    
      // ctx->RefCountIsOne() should be true here.
      tensorflow::unwrap(ctx)->Release();
    }
    
    TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) {
      TF_DeviceList* l = new TF_DeviceList;
      tensorflow::unwrap(ctx)->ListDevices(&l->response);
      return l;
    }
    
    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

    }
    
    void TFE_ContextEnableGraphCollection(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->SetShouldStoreGraphs(true);
    }
    
    void TFE_ContextDisableGraphCollection(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->SetShouldStoreGraphs(false);
    }
    
    uint64_t TFE_GetContextId(TFE_Context* ctx) {
      tensorflow::EagerContext* context =
          tensorflow::ContextFromInterface(tensorflow::unwrap(ctx));
      return context->GetContextId();
    }
    
    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