- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 17 for Int32 (0.12 sec)
-
tensorflow/c/experimental/gradients/nn_grad_test.cc
ASSERT_EQ(errors::OK, status_.code()) << status_.message(); X.reset(X_raw); } // Label int32_t Y_vals[] = {1, 0, 1}; int64_t Y_dims[] = {3}; AbstractTensorHandlePtr Y; { AbstractTensorHandle* Y_raw; status_ = TestTensorHandleWithDims<int32_t, TF_INT32>( immediate_execution_ctx_.get(), Y_vals, Y_dims, 1, &Y_raw); ASSERT_EQ(errors::OK, status_.code()) << status_.message();
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
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); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_DeleteTensor(t);
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) -
tensorflow/c/c_api_test.cc
TF_Tensor* out = csession.output_tensor(0); ASSERT_TRUE(out != nullptr); EXPECT_EQ(TF_INT32, TF_TensorType(out)); EXPECT_EQ(0, TF_NumDims(out)); // scalar ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); int32* output_contents = static_cast<int32*>(TF_TensorData(out)); EXPECT_EQ(3 + 2, *output_contents); // Add another operation to the graph.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/eager/dlpack_test.cc
std::vector<int64_t> shape, std::vector<int64_t> strides) { size_t num_elements = 1; for (int i = 0; i < static_cast<int32_t>(shape.size()); ++i) { num_elements *= shape[i]; } std::vector<float> data(num_elements); for (size_t j = 0; j < num_elements; ++j) { data[j] = j; } DLManagedTensor dlm_in = {};
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) -
tensorflow/c/c_api_function_test.cc
ASSERT_TRUE(out != nullptr); EXPECT_EQ(TF_INT32, TF_TensorType(out)); EXPECT_EQ(0, TF_NumDims(out)); // scalar ASSERT_EQ(sizeof(int32_t), TF_TensorByteSize(out)); int32_t* output_contents = static_cast<int32_t*>(TF_TensorData(out)); EXPECT_EQ(expected_results[i], *output_contents); } } void CompareInt32Tensor(const std::vector<int32_t>& expected, TF_Tensor* t) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6) -
tensorflow/c/eager/immediate_execution_context.h
// will be blocked till the copy completes. This is the default policy. DEVICE_PLACEMENT_SILENT = 2, // Placement policy which silently copies int32 tensors but not other dtypes. DEVICE_PLACEMENT_SILENT_FOR_INT32 = 3, }; // LINT.ThenChange(//tensorflow/c/eager/c_api.h) // Abstract interface to a context. // // A context is responsible for creating key objects such as Tensors,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
return th; } TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, int value) { int data[] = {value}; TF_Status* status = TF_NewStatus(); TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_INT32, 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) -
tensorflow/c/eager/gradient_checker.cc
namespace gradients { using namespace std; // ================== Helper functions ================= // Fills data with values [start,end) with given step size. void Range(vector<int32_t>* data, int32_t start, int32_t end, int32_t step = 1) { for (int32_t i = start; i < end; i += step) { (*data)[i] = i; } } // Fills out_dims with the dimensions of the given tensor. void GetDims(const TF_Tensor* t, int64_t* out_dims) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
case TF_DataType::TF_FLOAT: case TF_DataType::TF_DOUBLE: dtype.code = DLDataTypeCode::kDLFloat; break; case TF_DataType::TF_INT8: case TF_DataType::TF_INT16: case TF_DataType::TF_INT32: case TF_DataType::TF_INT64: dtype.code = DLDataTypeCode::kDLInt; break; case TF_DataType::TF_UINT8: case TF_DataType::TF_UINT16: case TF_DataType::TF_UINT32:
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) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
} auto pjrt_client = xla::GetCApiClient(DEVICE_CPU); CHECK_OK(pjrt_client.status()); auto c_api_client = down_cast<xla::PjRtCApiClient*>(pjrt_client->get()); std::vector<int32_t> data(1, 0); xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); auto buffer = c_api_client->pjrt_c_client()->client->BufferFromHostBuffer( data.data(), shape.element_type(), shape.dimensions(),
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0)