- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 90 for device (0.17 sec)
-
tensorflow/c/eager/custom_device_testutil.h
limitations under the License. ==============================================================================*/ #ifndef TENSORFLOW_C_EAGER_CUSTOM_DEVICE_TESTUTIL_H_ #define TENSORFLOW_C_EAGER_CUSTOM_DEVICE_TESTUTIL_H_ // A simple logging device to test custom device registration. #include <memory> #include "tensorflow/c/c_api.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h"
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 1.6K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.h
void AllocateParallelDevice(const char* device_name, const char* const* underlying_devices, int num_underlying_devices, TFE_CustomDevice* device, void** device_info); } // namespace parallel_device } // namespace tensorflow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jun 04 21:49:16 GMT 2020 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
const char* first_device = "/job:worker/replica:0/task:1/device:CPU:0"; const char* second_device = "/job:worker/replica:0/task:2/device:CPU:0"; const char* device_name = "/job:localhost/replica:0/task:0/device:CUSTOM:0"; std::array<const char*, 2> underlying_devices{first_device, second_device}; RegisterParallelDevice(context.get(), device_name, underlying_devices, status.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/parallel_device/parallel_device_testlib.h
} template <std::size_t num_devices> void RegisterParallelDevice( TFE_Context* context, const char* device_name, const std::array<const char*, num_devices>& underlying_devices, TF_Status* status) { TFE_CustomDevice device; void* device_info; tensorflow::parallel_device::AllocateParallelDevice( device_name, underlying_devices.data(), underlying_devices.size(), &device, &device_info);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
TFE_CustomDevice custom_device; custom_device.copy_tensor_to_device = &CopyToLoggingDevice; custom_device.copy_tensor_from_device = &CopyTensorFromLoggingDevice; custom_device.delete_device = &DeleteLoggingDevice; custom_device.execute = &LoggingDeviceExecute; LoggingDevice* device = new LoggingDevice; device->arrived_flag = arrived_flag; device->executed_flag = executed_flag;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/BUILD
# tensorflow/python:_pywrap_parallel_device. filegroup( name = "lib_headers", srcs = ["parallel_device_lib.h"], ) filegroup( name = "lib_sources", srcs = ["parallel_device_lib.cc"], ) filegroup( name = "device_headers", srcs = ["parallel_device.h"], ) filegroup( name = "device_sources", srcs = ["parallel_device.cc"], ) filegroup( name = "headers",
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 01 20:19:06 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.h
// devices. virtual Status Dim(int dim_index, int64_t* dim) const = 0; // Returns the device which created the handle. virtual const char* DeviceName(Status* status) const = 0; // Returns the device where the tensor was placed. virtual const char* BackingDeviceName(Status* status) const = 0; // Returns the device type which created the handle.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Mar 10 21:56:24 GMT 2023 - 4.3K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler_internal.h
}; using OwnedTFBuffer = std::unique_ptr<TF_Buffer, TFBufferDeleter>; class CGraphOptimizer : public CustomGraphOptimizer { public: explicit CGraphOptimizer(TP_Optimizer optimizer, const char* device_type) : optimizer_(optimizer), device_type_(device_type) { if (optimizer.create_func != nullptr) { c_optimizer_ = (*optimizer_.create_func)(); } else { c_optimizer_ = nullptr; } }
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jun 08 08:58:23 GMT 2022 - 3.5K bytes - Viewed (1) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
Tensor* tensor); absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type); absl::Status ResetPjRtClient(const DeviceType& device_type); } // namespace tensorflow
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 09 19:12:59 GMT 2023 - 1.5K bytes - Viewed (0) -
tensorflow/c/c_api_internal.h
// Set whether to propagate assigned device information when constructing a new // Graph from a GraphDef. By default assigned device information is not copied // and is re-computed by the runtime. inline void TF_ImportGraphDefOptionsSetPropagateDeviceSpec( TF_ImportGraphDefOptions* opts, unsigned char propagate_device_spec) { opts->opts.propagate_device_spec = propagate_device_spec; } } // end namespace tensorflow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sat May 13 00:49:12 GMT 2023 - 7.6K bytes - Viewed (0)