- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for count (0.16 sec)
-
tensorflow/c/c_api.cc
if (edge->src_output() == oper_out.index) { if (count < max_consumers) { consumers[count] = {ToOperation(edge->dst()), edge->dst_input()}; } ++count; } } return count; } int TF_OperationNumControlInputs(TF_Operation* oper) { int count = 0; for (const auto* edge : oper->node.in_edges()) { if (edge->IsControlEdge() && !edge->src()->IsSource()) {
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/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
tensorflow::Tensor tensor(allocator.get(), DT_FLOAT, {1}); TF_ASSERT_OK_AND_ASSIGN( auto pjrt_client, xla::GetTfrtCpuClient(/*asynchronous=*/true, /*cpu_device_count=*/1)); std::vector<int32_t> data(1, 0); xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); TF_ASSERT_OK_AND_ASSIGN( auto buffer, pjrt_client->BufferFromHostBuffer(
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/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/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/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc
memset(buffer, 'x', n); TF_SetStatus(status, TF_OK, ""); return n; }; const uint32 block_count = 256; tf_gcs_filesystem::RamFileBlockCache cache( block_size, block_count * block_size, 0, fetcher); std::vector<char> out; out.resize(block_count, 0); // The cache has space for `block_count` blocks. The loop with i = 0 should // fill the cache, and the loop with i = 1 should be all cache hits. The
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:57 GMT 2021 - 23.2K 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/checkpoint_reader.cc
new TensorSliceReader::VarToDataTypeMap); v2_reader_->Seek(kHeaderEntryKey); for (v2_reader_->Next(); v2_reader_->Valid(); v2_reader_->Next()) { if (filtered_keys.count(string(v2_reader_->key())) > 0) continue; CHECK(entry.ParseFromArray(v2_reader_->value().data(), v2_reader_->value().size())) << entry.InitializationErrorString();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
EXPECT_EQ(orig_ptr, TF_TensorData(t)); } else { // In sync mode, forwarding can't really happen since the client code will // have a reference count on the input tensor while the kernel is being // executed and thus it cannot be re-used for the return tensor. EXPECT_NE(orig_ptr, TF_TensorData(t)); } }
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
tensorflow::Flag("tmp_dir", tensorflow::SetTmpDir, "", "Temporary directory to store test data.")}; if (!tensorflow::Flags::Parse(&argc, argv, flag_list)) { std::cout << tensorflow::Flags::Usage(argv[0], flag_list); return -1; } tensorflow::testing::InstallStacktraceHandler(); tensorflow::ModularFileSystemTest::InitializeTestRNG();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 27 20:25:58 GMT 2022 - 71K bytes - Viewed (0)