Searched refs:CustomDevice (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.h | 34 class CustomDevice {
36 virtual ~CustomDevice() {} in ~CustomDevice()
59 using VariantDevice = absl::variant<Device*, CustomDevice*>;
75 CustomDevice* device, tensorflow::DataType dtype) in CustomDeviceTensorHandle()
95 CustomDevice* device() const { return device_; } in device()
116 CustomDevice* const device_;
|
| D | custom_device_op_handler.h | 30 std::unique_ptr<CustomDevice> device);
34 CustomDevice** device) const;
41 Status MaybePinToCustomDevice(CustomDevice** device,
47 std::unordered_map<string, std::unique_ptr<CustomDevice>> custom_devices_;
|
| D | custom_device_op_handler.cc | 25 const string& device_name, std::unique_ptr<CustomDevice> device) { in RegisterCustomDevice()
45 const string& name, CustomDevice** device) const { in FindCustomDeviceFromName()
57 tensorflow::CustomDevice* custom_device = nullptr; in Execute()
99 CustomDevice** device, const ImmediateExecutionOperation& op) const { in MaybePinToCustomDevice()
113 CustomDevice* first = nullptr; in MaybePinToCustomDevice()
121 CustomDevice* current = input->device(); in MaybePinToCustomDevice()
|
| D | custom_device_test.cc | 31 class TestCustomDevice : public CustomDevice {
86 TEST(CustomDevice, TestTensorHandle) { in TEST() argument
111 TEST(CustomDevice, TestResourcePlacement) { in TEST() argument
141 CustomDevice* placed_device = nullptr; in TEST()
|
| D | core.cc | 113 tensorflow::CustomDevice* dev; in CopyTensorHandleToDevice()
130 tensorflow::CustomDevice* dev; in CopyTensorHandleToDevice()
|
| D | eager_operation.cc | 478 } else if (absl::holds_alternative<CustomDevice*>(device)) { in VariantDeviceDebugString()
479 return absl::get<CustomDevice*>(device)->name(); in VariantDeviceDebugString()
|
| D | context.h | 480 std::unique_ptr<CustomDevice> device) override;
|
| D | context.cc | 908 const string& device_name, std::unique_ptr<CustomDevice> device) { in RegisterCustomDevice()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_context.h | 42 class CustomDevice; variable
164 std::unique_ptr<CustomDevice> device) = 0;
|
| D | c_api.cc | 409 class CustomDeviceAPI : public tensorflow::CustomDevice {
494 tensorflow::CustomDevice* device, in CAPICustomDeviceTensorHandle()
536 tensorflow::CustomDevice* device = nullptr; in TFE_NewCustomDeviceTensorHandle()
566 tensorflow::CustomDevice* custom_device = nullptr; in TFE_NewTensorHandleFromDeviceMemory()
|