- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TF_NewTensor (0.39 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
} void Variable::Assign(TFE_Context* context, TFE_TensorHandle* value, TF_Status* status) { GeneralAssignment("AssignVariableOp", context, value, status); } // Passed to `TF_NewTensor` to indicate how an array of floats should be // deleted. static void FloatDeallocator(void* data, size_t, void* arg) { delete[] static_cast<float*>(data); } // Creates a TFE_TensorHandle with value `v`.
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
int32_t* values = new int32_t[1]; values[0] = v; return TF_NewTensor(TF_INT32, nullptr, 0, values, num_bytes, &Int32Deallocator, nullptr); } TF_Tensor* DoubleTensor(double v) { const int num_bytes = sizeof(double); double* values = new double[1]; values[0] = v; return TF_NewTensor(TF_DOUBLE, nullptr, 0, values, num_bytes, &DoubleDeallocator, nullptr); }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
++device_index) { auto device_value = absl::make_unique<DataType>(); *device_value = values[device_index]; std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> tensor( TF_NewTensor( datatype_enum, /*dims=*/nullptr, /*num_dims=*/0, device_value.release(), sizeof(DataType), [](void* data, size_t, void* arg) { delete reinterpret_cast<DataType*>(data);
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw( TF_TensorDefaultAlignment(), num_bytes)); int64_t dims[] = {2, 3}; bool deallocator_called = false; TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, &Deallocator, &deallocator_called); EXPECT_FALSE(deallocator_called); EXPECT_EQ(TF_FLOAT, TF_TensorType(t)); EXPECT_EQ(2, TF_NumDims(t));
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/c_api.cc
dims.reserve(shape.dims()); for (int i = 0; i < shape.dims(); ++i) { dims.push_back(shape.dim_size(i)); nelems *= shape.dim_size(i); } CHECK_EQ(nelems, 0); return TF_NewTensor( dtype, reinterpret_cast<const int64_t*>(dims.data()), shape.dims(), reinterpret_cast<void*>(&empty), 0, [](void*, size_t, void*) {}, nullptr); } static void TF_Run_Helper(
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0)