- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for once (0.44 sec)
-
tensorflow/c/c_api.cc
body_fn, params->name, &loop_outputs); // Update name_map with newly-created ops. // TODO(skyewm): right now BuildWhileLoop() may alter the graph if it returns // a bad status. Once we fix this, we may want to return early instead of // executing the following code. for (int i = first_new_node_id; i < parent->graph.num_node_ids(); ++i) { Node* new_node = parent->graph.FindNodeId(i);
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/parallel_device/parallel_device_remote_test.cc
TensorHandlePtr value_one(FloatTensorHandle(3., status.get())); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TensorHandlePtr value_two(FloatTensorHandle(-2., status.get())); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); std::array<TFE_TensorHandle*, 2> in_components{value_one.get(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
} private: bool first_call_ = true; }; // After the graph pass is registered, it takes effect globally and can affect // other test cases. Define a static variable to switch it on and off. bool GraphErrorInjectionPass::enabled_ = false; // Test to ensure that a registered graph optimization pass is only executed
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.cc
auto& indices = iter->second; if (std::find(indices.begin(), indices.end(), idx) != indices.end()) { return InvalidArgument("TF_Output ", node->name(), ":", idx, " appears more than once in the input list"); } indices.push_back(idx); } } return absl::OkStatus(); } // Converts `noutputs` and `outputs` into `outputs_tensors` and does various // checks while doing so.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// // TODO(allenl): There are some use-cases that are only supported by copying to // host at the moment (e.g. debug print on a tensor, .numpy(), etc.). We either // need to return something here or address these use-cases one by one. TFE_TensorHandle* CopyTensorFromParallelDevice(TFE_Context* context, TFE_TensorHandle* tensor, const char* target_device_name,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
// Outputs: std::vector<TensorHandlePtr> op_outputs_ TF_GUARDED_BY(execution_mutex_); // TF_Status is an incomplete type and so can't be stack allocated. To avoid // unnecessary allocations each Execute call, we keep one heap-allocated // version for the thread. StatusPtr status_ TF_GUARDED_BY(execution_mutex_); const std::string device_; ExecutorPtr executor_ TF_GUARDED_BY(execution_mutex_);
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/experimental/filesystem/modular_filesystem_registration.cc
TF_RETURN_IF_ERROR(ValidatePluginMemoryRoutines(info)); // Validate and register all filesystems // Try to register as many filesystems as possible. // Free memory once we no longer need it Status status; for (int i = 0; i < info->num_schemes; i++) { status.Update(ValidateAndRegisterFilesystems(info, i)); info->plugin_memory_free(info->ops[i].scheme);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 07 22:08:43 GMT 2023 - 12.8K bytes - Viewed (0) -
tensorflow/c/c_api_function_test.cc
params->body_graph, s_, "add1"); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); TF_Operation* one = ScalarConst(1, params->body_graph, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); TF_Operation* add2 = Add(add1, one, params->body_graph, s_, "add2"); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); params->body_outputs[0] = {add2, 0};
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/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
constexpr char kMaxCacheSize[] = "GCS_READ_CACHE_MAX_SIZE_MB"; constexpr size_t kDefaultMaxCacheSize = 0; // The environment variable that overrides the maximum staleness of cached file // contents. Once any block of a file reaches this staleness, all cached blocks // will be evicted on the next read. constexpr char kMaxStaleness[] = "GCS_READ_CACHE_MAX_STALENESS"; constexpr uint64_t kDefaultMaxStaleness = 0;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 23 06:55:53 GMT 2023 - 46.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
tensorflow::ImmediateExecutionTensorHandle* unwrapped_handle = tensorflow::unwrap(handles[i]); if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { // One of the inputs we're trying to pack is on a custom device. We'll let // the first custom device we see handle all of the packing. auto* custom_device_handle =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3)