- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for arg (0.14 sec)
-
tensorflow/c/eager/c_api.cc
void (*deallocator)(void* data, size_t len, void* arg), void* deallocator_arg, TF_Status* status) { tensorflow::Device* device = nullptr; tensorflow::EagerContext* context = tensorflow::ContextFromInterface(tensorflow::unwrap(ctx)); status->status = context->FindDeviceFromName(device_name, &device); if (!status->status.ok()) { deallocator(data, len, deallocator_arg); status->status =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_test.cc
ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); auto cpu_arg = tensorflow::TensorHandleFromInterface(tensorflow::unwrap(hcpu)); auto gpu_arg = tensorflow::TensorHandleFromInterface(tensorflow::unwrap(hgpu)); auto gpu_device = gpu_arg->device(); ASSERT_FALSE(cpu_arg->HasLocalMirror(gpu_device)); TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); if (cpu_op) {
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/experimental/filesystem/modular_filesystem_test.cc
// // Example invocation: // bazel test //tensorflow/c/experimental/filesystem:modular_filesystem_test \\ // --test_arg=--dso=/path/to/one.so --test_arg=--dso=/path/to/another.so \\ // --test_arg=--scheme= --test_arg=--scheme=file // // Note that to test the local filesystem we use an empty value. namespace tensorflow { namespace { using ::tensorflow::error::Code;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 27 20:25:58 GMT 2022 - 71K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
// Build two abstract input tensors as function arguments. std::vector<TF_AbstractTensor*> func_args; { TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(eager_execution_ctx, s); // 1st Arg float vals1[] = {0.0f, 1.0f, 1.0f, 0.0f}; int64_t dims[] = {2, 2}; // Matrices will be 2 x 2 int num_dims = sizeof(dims) / sizeof(dims[0]); TFE_TensorHandle* input_eager =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
EXPECT_EQ(TF_CANCELLED, TF_GetCode(s)); EXPECT_EQ(string("cancel"), TF_Message(s)); TF_DeleteStatus(s); } void Deallocator(void* data, size_t, void* arg) { tensorflow::cpu_allocator()->DeallocateRaw(data); *reinterpret_cast<bool*>(arg) = true; } TEST(CAPI, Tensor) { const int num_bytes = 6 * sizeof(float); float* values = reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw(
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)