- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for Kata (0.13 sec)
-
tensorflow/c/eager/dlpack_test.cc
dltensor_out->shape[i + 1] * dltensor_out->strides[i + 1]); } } } const float* data_in = static_cast<const float*>(dltensor_in->data); const float* data_out = static_cast<const float*>(dltensor_out->data); for (size_t j = 0; j < num_elements; ++j) { EXPECT_EQ(data_out[j], data_in[j]); } TFE_CallDLManagedTensorDeleter(dlm_out); TFE_DeleteTensorHandle(handle); }
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/experimental/gradients/nn_grad.cc
*/ AbstractTensorHandle* upstream_grad = grad_outputs[0]; DCHECK(upstream_grad); // Recover data format from forward pass for gradient. std::string data_format; TF_RETURN_IF_ERROR(forward_attrs_.Get("data_format", &data_format)); // Grad for A grad_inputs[0] = upstream_grad; grad_inputs[0]->Ref(); // Grad for bias
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
server_def, tensorflow::Env::Default(), &worker_server2) .ok()); ASSERT_TRUE(worker_server2->Start().ok()); TFE_ContextSetServerDef(context.get(), 0, serialized.data(), serialized.size(), status.get()); EXPECT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); BasicTestsForTwoDevices(context.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) -
tensorflow/c/eager/c_api_unified_experimental.cc
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test.cc
TFE_Context* ctx = TFE_NewContext(opts, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); 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);
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) -
tensorflow/c/env_test.cc
bool did_work = false; }; void SomeThreadFunc(void* data) { auto* real_data = static_cast<SomeThreadData*>(data); ::tensorflow::mutex_lock l(real_data->mu); real_data->did_work = true; } } // namespace TEST(TestEnv, TestThreads) { TF_ThreadOptions options; TF_DefaultThreadOptions(&options); SomeThreadData data; TF_Thread* thread =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Dec 10 20:52:48 GMT 2018 - 4.2K bytes - Viewed (0) -
tensorflow/c/env.cc
} void TF_AppendWritableFile(TF_WritableFileHandle* handle, const char* data, size_t length, TF_Status* status) { auto* cc_file = reinterpret_cast<::tensorflow::WritableFile*>(handle); TF_SetStatus(status, TF_OK, ""); ::tensorflow::Set_TF_Status_from_Status( status, cc_file->Append(::tensorflow::StringPiece{data, length})); } void TF_DeleteFile(const char* filename, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 11 01:20:50 GMT 2021 - 7K bytes - Viewed (0) -
tensorflow/c/eager/gradient_checker.cc
GetDims(theta_tensor, theta_dims.data()); // Initialize auxilary data structures. vector<float> thetaPlus_data(num_elems); vector<float> thetaMinus_data(num_elems); AbstractTensorHandle* f_outputs[1]; // Numerical Grad Check for (int i = 0; i < num_elems; i++) { // Get relative epsilon value float epsilon = theta_data[i] == 0 ? 1e-4 : std::abs(theta_data[i] * 1e-4);
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/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
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(), /*byte_strides=*/std::nullopt,
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) -
tensorflow/c/eager/custom_device_testutil.cc
}; int64_t LoggedTensorDim(void* data, int dim_index, TF_Status* status) { return TFE_TensorHandleDim(reinterpret_cast<LoggedTensor*>(data)->tensor, dim_index, status); } int LoggedTensorNumDims(void* data, TF_Status* status) { return TFE_TensorHandleNumDims(reinterpret_cast<LoggedTensor*>(data)->tensor, status); }
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)