- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for argv (0.14 sec)
-
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/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) -
tensorflow/c/c_api_function_test.cc
// Verify that FunctionDef ArgDef has attributes. ASSERT_EQ(func_->record->fdef().arg_attr_size(), 1); auto arg_attrs = func_->record->fdef().arg_attr().find(0); ASSERT_NE(arg_attrs, func_->record->fdef().arg_attr().end()); auto iter = arg_attrs->second.attr().find("_test_attr"); ASSERT_NE(iter, arg_attrs->second.attr().end()); EXPECT_EQ(iter->second.s(), "value"); }
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) -
configure.py
parser.add_argument( '--workspace', type=str, default=os.path.abspath(os.path.dirname(__file__)), help='The absolute path to your active Bazel workspace.') args = parser.parse_args() _TF_WORKSPACE_ROOT = args.workspace _TF_BAZELRC = os.path.join(_TF_WORKSPACE_ROOT, _TF_BAZELRC_FILENAME) # Make a copy of os.environ to be clear when functions and getting and setting # environment variables.
Python - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 18:25:36 GMT 2024 - 53.8K bytes - Viewed (1) -
tensorflow/c/c_api.h
// are not optimizing for convenience. // * We make assumption that one session has one graph. This should be // fine since we have the ability to run sub-graphs. // * We could allow NULL for some arguments (e.g., NULL options arg). // However since convenience is not a primary goal, we don't do this. // * Devices are not in this API. Instead, they are created/used internally // and the API just provides high level controls over the number of
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3)