- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 27 for tool (0.39 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_test.cc
std::unique_ptr<TF_DeviceList, decltype(&TF_DeleteDeviceList)> devices( TFE_ContextListDevices(context.get(), status.get()), TF_DeleteDeviceList); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); bool has_tpu = false; for (int device_index = 0; device_index < TF_DeviceListCount(devices.get()); ++device_index) { std::string device_type = TF_DeviceListType(devices.get(), device_index, status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
// deadlocks. class DeviceThread { public: // Starts a background thread waiting for `StartExecute`. explicit DeviceThread(const std::string& device, const bool is_async, const int in_flight_nodes_limit) : status_(TF_NewStatus()), // If the context's default exector is set to async, re-using that in
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/next_pluggable_device/c_api.cc
TF_OpKernelContext* ctx) { auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelContext*>(ctx); return reinterpret_cast<TF_CoordinationServiceAgent*>( cc_ctx->coordination_service_agent()); } bool TF_CoordinationServiceIsInitialized(TF_CoordinationServiceAgent* agent) { if (agent == nullptr) return false; auto* cc_agent = reinterpret_cast<tsl::CoordinationServiceAgent*>(agent); return cc_agent->IsInitialized(); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/c_api_function.cc
} TF_Function* TF_FunctionImportFunctionDef(const void* proto, size_t proto_len, TF_Status* status) { tensorflow::FunctionDef fdef; bool success = fdef.ParseFromArray(proto, proto_len); if (!success) { status->status = InvalidArgument( "Invalid FunctionDef given to TF_FunctionImportFunctionDef"); return nullptr; }
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
std::move(maybe_typed_outputs.value())); if (typed_outputs.size() > *num_outputs) { TF_SetStatus(status, TF_INTERNAL, "The allocated output buffer was too small."); return; } for (int i = 0; i < typed_outputs.size(); ++i) { MaybeParallelTensorOwned typed_output(std::move(typed_outputs[i])); if (absl::holds_alternative<TensorHandlePtr>(typed_output)) {
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/c_api_unified_experimental_graph.cc
int num_values) override { std::unique_ptr<bool[]> b(new bool[num_values]); for (int i = 0; i < num_values; ++i) { b[i] = values[i]; } op_->node_builder.Attr(attr_name, ArraySlice<const bool>(b.get(), num_values)); return absl::OkStatus(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
typedef struct GCSFile { const std::string path; const bool is_cache_enable; const uint64_t buffer_size; ReadFn read_fn; absl::Mutex buffer_mutex; uint64_t buffer_start ABSL_GUARDED_BY(buffer_mutex); bool buffer_end_is_past_eof ABSL_GUARDED_BY(buffer_mutex); std::string buffer ABSL_GUARDED_BY(buffer_mutex); GCSFile(std::string path, bool is_cache_enable, uint64_t buffer_size, ReadFn read_fn)
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
} TFE_Executor* TFE_NewExecutor(bool is_async, bool enable_streaming_enqueue, int in_flight_nodes_limit) { return new TFE_Executor(is_async, enable_streaming_enqueue, in_flight_nodes_limit); } void TFE_DeleteExecutor(TFE_Executor* executor) { delete executor; } bool TFE_ExecutorIsAsync(TFE_Executor* executor) {
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) -
tensorflow/c/eager/c_api.cc
// Set server def with timeout. TF_CAPI_EXPORT extern void TFE_ContextSetServerDefWithTimeout( TFE_Context* ctx, int keep_alive_secs, const void* proto, size_t proto_len, int64_t init_timeout_in_ms, TF_Status* status, bool clear_existing_contexts) { TFE_ContextSetServerDefWithTimeoutAndRetries( ctx, keep_alive_secs, proto, proto_len, init_timeout_in_ms, /*retries=*/0, status, clear_existing_contexts); }
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) -
tensorflow/c/eager/c_api_test_util.cc
TF_DeleteTensor(t); TF_DeleteStatus(status); return th; } TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, bool value) { bool data[] = {value}; TF_Status* status = TF_NewStatus(); TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_BOOL, nullptr, 0, status); memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); TFE_TensorHandle* th = TFE_NewTensorHandleFromTensor(ctx, t, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2)