- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 25 for cout (0.18 sec)
-
tensorflow/c/c_api.cc
int TF_OperationOutputNumConsumers(TF_Output oper_out) { int count = 0; for (const auto* edge : oper_out.oper->node.out_edges()) { if (edge->src_output() == oper_out.index) { ++count; } } return count; } int TF_OperationOutputConsumers(TF_Output oper_out, TF_Input* consumers, int max_consumers) { int count = 0;
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/c_api_unified_experimental.cc
return *factories; } static tracing::FactoryFunction default_factory; void RegisterTracingEngineFactory(const string& name, FactoryFunction factory) { assert((!GetFactories().count(name)) || (GetFactories()[name] == factory) && "Duplicate tracing factory registration"); GetFactories()[name] = factory; } Status SetDefaultTracingEngine(const char* name) {
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_cluster_test.cc
TFE_DeleteExecutor(executor); TF_DeleteStatus(status); } // Read the value of variable `var` and save it into `out_value`. void ReadVariable(TFE_Context* ctx, TFE_TensorHandle* var, TFE_TensorHandle** out_value) { TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
} auto* gpu_options = config.mutable_gpu_options(); gpu_options->set_allow_growth(gpu_memory_allow_growth); (*config.mutable_device_count())["CPU"] = num_cpu_devices; // TODO(b/113217601): This is needed for EagerContext::runner_ to use a // threadpool, so that we avoid the possibility of running the runner_ in the
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/eager/c_api_remote_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_DeleteExecutor(executor); TFE_DeleteContext(ctx); TF_DeleteStatus(status); // TODO(b/136478427): Figure out how to correctly shut the server down. worker_server.release(); } TEST(CAPI, RemoteExecute) { TestRemoteExecute(false); } TEST(CAPI, RemoteExecuteAsync) { TestRemoteExecute(true); }
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/experimental/gradients/math_grad_test.cc
ExpModel, BuildGradModel(ExpModel, registry_), immediate_execution_ctx_.get(), {x.get()}, UseFunction())); } TEST_P(CppGradients, TestMatMulGrad) { // TODO(vnvo2409): Figure out why `gradient_checker` does not work very // well with `MatMul` and remove `TestMatMul*` in // `mnist_gradients_test` when done. GTEST_SKIP();
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc
if (r > 0) { dst += r; offset += static_cast<uint64_t>(r); n -= r; // safe as 0 < r <= n so n will never underflow read += r; } else if (r == 0) { TF_SetStatus(status, TF_OUT_OF_RANGE, "Read fewer bytes than requested"); break; } else if (errno == EINTR || errno == EAGAIN) { // Retry } else { TF_SetStatusFromIOError(status, errno, posix_file->filename); break;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/env.cc
void TF_DeleteRecursively(const char* dirname, int64_t* undeleted_file_count, int64_t* undeleted_dir_count, TF_Status* status) { ::int64_t f, d; TF_SetStatus(status, TF_OK, ""); ::tensorflow::Set_TF_Status_from_Status( status, ::tensorflow::Env::Default()->DeleteRecursively(dirname, &f, &d)); *undeleted_file_count = f; *undeleted_dir_count = d; }
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/parallel_device/parallel_device_lib.cc
TF_Message(status)); } if (device_index == 0) { first_op_output_count = per_device_output_tensors.rbegin()->size(); } else { if (first_bad_status == nullptr && per_device_output_tensors.rbegin()->size() != first_op_output_count) { first_bad_status.reset(TF_NewStatus()); TF_SetStatus(first_bad_status.get(), TF_INTERNAL,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/gradient_checker.cc
AbstractTensorHandlePtr model_out(model_outputs[0]); TF_Tensor* model_out_tensor; TF_RETURN_IF_ERROR(GetValue(model_out.get(), &model_out_tensor)); int num_dims_out = TF_NumDims(model_out_tensor); TF_DeleteTensor(model_out_tensor); // If the output is a scalar, then return the scalar output if (num_dims_out == 0) { outputs[0] = model_out.release(); return absl::OkStatus(); }
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)