Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for to_tensor (0.07 sec)

  1. tensorflow/c/eager/c_api_test_util.cc

      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
      TFE_OpAddInput(op, var_handle, status);
    
      // Convert 'value' to a TF_Tensor then a TFE_TensorHandle.
      std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> t(
          TF_AllocateTensor(TF_FLOAT, nullptr, 0, sizeof(value)), TF_DeleteTensor);
      memcpy(TF_TensorData(t.get()), &value, TF_TensorByteSize(t.get()));
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  2. tensorflow/c/c_test_util.cc

    static void FloatDeallocator(void* data, size_t, void* arg) {
      delete[] static_cast<float*>(data);
    }
    
    TF_Tensor* BoolTensor(bool v) {
      const int num_bytes = sizeof(bool);
      bool* values = new bool[1];
      values[0] = v;
      return TF_NewTensor(TF_BOOL, nullptr, 0, values, num_bytes, &BoolDeallocator,
                          nullptr);
    }
    
    TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values) {
      int64_t num_values = 1;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  3. tensorflow/c/BUILD

        name = "tf_tensor_hdrs",
        hdrs = [
            "tf_tensor.h",
            "tf_tensor_helper.h",
        ],
        visibility = ["//visibility:public"],
        deps = [
            ":c_api_macros_hdrs",
            ":tf_datatype_hdrs",
            ":tf_status_headers",
            "//tensorflow/core/platform:status",
        ],
    )
    
    cc_library(
        name = "tf_tensor",
        srcs = ["tf_tensor.cc"],
        hdrs = [
            "tf_tensor.h",
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Nov 02 06:47:06 UTC 2024
    - 30.4K bytes
    - Viewed (0)
  4. tensorflow/c/eager/dlpack.cc

        return nullptr;
      }
      TF_DataType dtype;
      absl::Status s = TfDataTypeFormDlDataType(dl_tensor->dtype, &dtype);
      if (!s.ok()) {
        status->status = std::move(s);
        return nullptr;
      }
      int num_dims = dl_tensor->ndim;
      const int64_t* dims = dl_tensor->shape;
      void* data = dl_tensor->data;
    
      if (dl_tensor->byte_offset != 0) {
        status->status = tensorflow::errors::InvalidArgument(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental_test.cc

    }
    
    TEST_F(ShapeInferenceTest, InfersShapesFromInputTensors) {
      // Prepare some tensors for shape.
      TF_Tensor* tensor_1X6 = Int32Tensor({1, 6});
      CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);
      TF_Tensor* tensor_1X1X6 = Int32Tensor({1, 1, 6});
      CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Jan 17 22:27:52 UTC 2023
    - 13.1K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.h

    // A handle to a tensor on a device.
    //
    // Like a TF_Tensor, a TFE_TensorHandle refers to a tensor with a value, shape,
    // type etc. Unlike a TF_Tensor, a TFE_TensorHandle may refer to such tensors
    // placed in the memory of different devices or remote address spaces.
    typedef struct TFE_TensorHandle TFE_TensorHandle;
    
    TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandle(const TF_Tensor* t,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_experimental.cc

            binary_proto_buf.data(), binary_proto_buf.size(), status);
        if (!status->status.ok()) return {};
        ret.push_back(UniqueFuncPtr(func, TF_DeleteFunction));
      }
      return ret;
    }
    
    TF_Tensor* TF_DequeueNamedTensor(TF_Session* session, int tensor_id,
                                     TF_Status* status) {
      assert(session);
      {
        tensorflow::mutex_lock c(session->graph->mu);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  8. tensorflow/c/c_api_experimental.h

    // empty, this call is blocked.
    //
    // Tensors are enqueued via the corresponding TF enqueue op.
    // TODO(hongm): Add support for `timeout_ms`.
    TF_CAPI_EXPORT extern TF_Tensor* TF_DequeueNamedTensor(TF_Session* session,
                                                           int tensor_id,
                                                           TF_Status* status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_unified_experimental_test.cc

    #include "tensorflow/c/eager/c_api_unified_experimental_internal.h"
    #include "tensorflow/c/tf_datatype.h"
    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/c/tf_status_helper.h"
    #include "tensorflow/c/tf_tensor.h"
    #include "tensorflow/core/framework/full_type.pb.h"
    #include "tensorflow/core/platform/errors.h"
    #include "tensorflow/core/platform/status.h"
    #include "tensorflow/core/platform/test.h"
    
    using tensorflow::Status;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 39.1K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_experimental_test.cc

            /*in_flight_nodes_limit=*/0);
        TFE_ContextSetExecutorForThread(ctx, executor);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
        TF_Tensor* t =
            TF_AllocateTensor(TF_INT32, nullptr, 0, 1 * sizeof(tensorflow::int32));
        *reinterpret_cast<tensorflow::int32*>(TF_TensorData(t)) = 42;
        TFE_TensorHandle* h = TFE_NewTensorHandle(t, status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 03:14:26 UTC 2023
    - 31.5K bytes
    - Viewed (0)
Back to top