- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for argc (0.13 sec)
-
tensorflow/c/c_api_experimental.cc
} return op_reg_data->op_def.is_stateful(); } void TF_InitMain(const char* usage, int* argc, char*** argv) { tensorflow::port::InitMain(usage, argc, argv); } int TF_PickUnusedPortOrDie() { return tensorflow::internal::PickUnusedPortOrDie(); } TFE_TensorHandle* TFE_NewTensorHandleFromScalar(TF_DataType data_type, void* data, size_t len,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/c_api.cc
.NewSubScope(child_scope_name); if (dx != nullptr) { std::vector<tensorflow::Output> dx_arg = OutputsFromTFOutputs(dx, ny); status->status = AddSymbolicGradients(scope, y_arg, x_arg, dx_arg, &dy_arg); } else { status->status = AddSymbolicGradients(scope, y_arg, x_arg, &dy_arg); } // Update g->name_map with the name_map from the scope, which will contain
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
if (!status->status.ok()) { return nullptr; } return tensor; } // Deleter for DLManagedTensor void DLManagedTensorDeleter(DLManagedTensor* arg) { TfDlManagedTensorCtx* owner = static_cast<TfDlManagedTensorCtx*>(arg->manager_ctx); owner->reference.Unref(); delete owner; } // Converts TF_DATAType to DLPack data type. DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) {
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/eager/c_api_experimental_test.cc
// First create a simple identity function. TF_Graph* function_graph = TF_NewGraph(); TF_OperationDescription* arg_descr = TF_NewOperation(function_graph, "Placeholder", "arg"); TF_SetAttrType(arg_descr, "dtype", TF_INT32); TF_Status* status = TF_NewStatus(); TF_Operation* arg = TF_FinishOperation(arg_descr, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_OperationDescription* id_descr =
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/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/eager/c_api_distributed_test.cc
CHECK(tensorflow::protobuf::TextFormat::ParseFromString( " signature {" " name: 'AddVariablesFunction'" " input_arg {" " name: 'var'" " type: DT_RESOURCE" " }" " output_arg {" " name: 'sum'" " type: DT_FLOAT" " }" " }" " node_def {" " name: 'read0'"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
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) -
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)