Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 27 for TensorHandle (0.17 sec)

  1. tensorflow/c/eager/c_api_debug_test.cc

      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_Context* ctx = TFE_NewContext(opts, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      TFE_TensorHandle* h = TestScalarTensorHandle(ctx, 1.0f);
      TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 06 22:10:09 GMT 2020
    - 2.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_cluster_test.cc

      const char dev1_name[] = "/job:localhost/replica:0/task:1/device:CPU:0";
    
      TFE_TensorHandle* var_handle0 = TestVariable(ctx, 1.0, dev0_name);
      EXPECT_NE(var_handle0, nullptr);
      TFE_TensorHandle* var_handle1 = TestVariable(ctx, 2.0, dev1_name);
      EXPECT_NE(var_handle1, nullptr);
    
      TFE_TensorHandle* value_handle = nullptr;
      ReadVariable(ctx, var_handle1, &value_handle);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental_eager.cc

      }
      return wrap(static_cast<AbstractContext*>(unwrap(c_ctx)));
    }
    
    TF_AbstractTensor* TF_CreateAbstractTensorFromEagerTensor(TFE_TensorHandle* t,
                                                              TF_Status* s) {
      return wrap(static_cast<AbstractTensorHandle*>(unwrap(t)));
    }
    
    TFE_TensorHandle* TF_AbstractTensorGetEagerTensor(TF_AbstractTensor* at,
                                                      TF_Status* s) {
    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)
  4. tensorflow/c/eager/c_api_remote_test.cc

      TFE_DeleteContextOptions(opts);
    
      TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx);
      TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(ctx);
      const char remote_device_name[] =
          "/job:localhost/replica:0/task:1/device:CPU:0";
      auto* h0_task1 =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 12 00:14:22 GMT 2020
    - 5.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/dlpack_test.cc

      dltensor_in->device = {kDLCPU, 0};
      dltensor_in->ndim = static_cast<int32_t>(shape.size());
      dltensor_in->dtype = {kDLFloat, 32, 1};
      dltensor_in->shape = shape.data();
      dltensor_in->strides = strides.data();
      TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx);
      ASSERT_NE(handle, nullptr)
          << TF_Message(status) << " (shape=[" << absl::StrJoin(shape, ",")
          << "], strides=[" << absl::StrJoin(strides, ",") << "])";
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jun 30 03:04:46 GMT 2023
    - 4.4K bytes
    - Viewed (0)
  6. tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc

      TensorHandlePtr value_two(FloatTensorHandle(-2., status.get()));
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      std::array<TFE_TensorHandle*, 2> in_components{value_one.get(),
                                                     value_two.get()};
      TensorHandlePtr combined_value = CreatePerDeviceValues(
          context.get(), in_components, device_name, status.get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 22:09:57 GMT 2023
    - 6.7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/custom_device_test.cc

                            &arrived, &executed, status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_TensorHandle* hcpu = TestMatrixTensorHandle(context);
      ASSERT_FALSE(arrived);
      TFE_TensorHandle* hdevice =
          TFE_TensorHandleCopyToDevice(hcpu, context, name, status.get());
      ASSERT_TRUE(arrived);
      ASSERT_FALSE(executed);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  8. tensorflow/c/eager/dlpack.cc

    // Gets tensor from eager tensor handle.
    const Tensor* GetTensorFromHandle(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
      tensorflow::TensorHandle* handle =
          tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h));
      if (handle->Type() != TensorHandle::LOCAL) {
        status->status = tensorflow::errors::InvalidArgument(
    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)
  9. tensorflow/c/c_api_experimental.cc

    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/c/eager/c_api_internal.h"
    #include "tensorflow/c/eager/tfe_context_internal.h"
    #include "tensorflow/c/eager/tfe_op_internal.h"
    #include "tensorflow/c/eager/tfe_tensorhandle_internal.h"
    #include "tensorflow/c/tf_buffer_internal.h"
    #include "tensorflow/compiler/jit/flags.h"
    #include "tensorflow/core/common_runtime/eager/attr_builder.h"
    #include "tensorflow/core/common_runtime/eager/context.h"
    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)
  10. tensorflow/c/eager/custom_device_testutil.cc

                                             status);
    }
    
    TFE_TensorHandle* CopyToLoggingDevice(TFE_Context* context,
                                          TFE_TensorHandle* tensor,
                                          TF_Status* status, void* device_info) {
      LoggingDevice* dev = reinterpret_cast<LoggingDevice*>(device_info);
      TFE_TensorHandle* t = TFE_TensorHandleCopyToDevice(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
Back to top