- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 46 for TensorHandle (0.63 sec)
-
tensorflow/c/eager/tfe_tensorhandle_internal.h
// the TFE_TensorHandle structure. typedef struct TFE_TensorHandle TFE_TensorHandle; namespace tensorflow { DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionTensorHandle, TFE_TensorHandle); DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionTensorHandle*, TFE_TensorHandle*); } // namespace tensorflow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jun 17 19:39:13 GMT 2020 - 1.6K bytes - Viewed (0) -
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,
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) -
tensorflow/c/eager/c_api_test_util.h
// on the given device. TFE_TensorHandle* TestVariable(TFE_Context* ctx, float value, const tensorflow::string& device_name = ""); // Return an add op multiplying `a` by `b`. TFE_Op* AddOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b); // Return a matmul op multiplying `a` by `b`. TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b); // Return an identity op.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
using tensorflow::string; 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);
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) -
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) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
TFE_Context* context, const std::array<TFE_TensorHandle*, num_replicas>& components, const char* device, TF_Status* status); TensorHandlePtr Multiply(TFE_Context* context, TFE_TensorHandle* first, TFE_TensorHandle* second, TF_Status* status); // Assert that `handle` is equal to `expected_value`. template <typename value_type> void ExpectScalarEq(TFE_TensorHandle* handle, value_type expected_value);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
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) -
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) -
tensorflow/c/eager/c_api_test.cc
TFE_DeleteContextOptions(opts); TFE_TensorHandle* input1 = TestMatrixTensorHandle(ctx); TFE_TensorHandle* input2 = TestMatrixTensorHandle(ctx); TFE_TensorHandle* dim = TestScalarTensorHandle(ctx, 0); TFE_Op* concatOp = TFE_NewOp(ctx, "Concat", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* inputs[] = {input1, input2}; TFE_OpAddInput(concatOp, dim, status);
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) -
tensorflow/c/eager/c_api_experimental.h
// returned will be on the device in which the TFE_TensorHandle resides (so e.g. // for a GPU tensor this will return a pointer to GPU memory). The pointer is // only guaranteed to be valid until TFE_DeleteTensorHandle is called on this // TensorHandle. Only supports POD data types. TF_CAPI_EXPORT extern void* TFE_TensorHandleDevicePointer(TFE_TensorHandle*, TF_Status*);
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)