Searched refs:TestCustomDeviceTensorHandle (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | custom_device_test.cc | 67 class TestCustomDeviceTensorHandle : public CustomDeviceTensorHandle { class
69 TestCustomDeviceTensorHandle(ImmediateExecutionContext* context, in TestCustomDeviceTensorHandle() function in tensorflow::eager::__anonb9701f0e0111::TestCustomDeviceTensorHandle
106 core::RefCountPtr<TestCustomDeviceTensorHandle> tensor( in TEST()
107 new TestCustomDeviceTensorHandle(ctx.get(), &device, DT_FLOAT, in TEST()
135 core::RefCountPtr<TestCustomDeviceTensorHandle> tensor( in TEST()
136 new TestCustomDeviceTensorHandle(ctx.get(), &device, DT_FLOAT, in TEST()
154 core::RefCountPtr<TestCustomDeviceTensorHandle> custom_float_tensor( in TEST()
155 new TestCustomDeviceTensorHandle(ctx.get(), &custom_device, DT_FLOAT, in TEST()
157 core::RefCountPtr<TestCustomDeviceTensorHandle> custom_resource_tensor( in TEST()
158 new TestCustomDeviceTensorHandle(ctx.get(), &custom_device, DT_RESOURCE, in TEST()
|