Searched refs:device_id_ (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.h | 73 device_id_(device_id) { in MatmulParameters()
96 dtype_, ", ", device_id_); in ToString()
105 return std::make_tuple(transa_, transb_, m_, n_, k_, dtype_, device_id_); in get_data_as_tuple()
114 int device_id_; variable
|
| D | conv_ops_gpu.h | 118 device_id_(device_id), in batch_()
155 device_id_, in ToString()
185 device_id_, group_count_); in get_data_as_tuple()
221 int device_id_; variable
|
| D | bias_op.cc | 427 device_id_(device_id) { in BiasAddParams()
448 data_format_, ", ", dtype_, ", ", device_id_); in ToString()
456 return std::make_tuple(in_shape_, data_format_, dtype_, device_id_); in get_data_as_tuple()
465 int device_id_; member in tensorflow::BiasAddParams
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_program.cc | 105 : program_(program), device_id_(device_id) {} in CLProgram()
108 : program_(program.program_), device_id_(program.device_id_) { in CLProgram()
116 std::swap(device_id_, program.device_id_); in operator =()
|
| D | cl_program.h | 68 cl_device_id GetDeviceId() const { return device_id_; } in GetDeviceId()
78 cl_device_id device_id_ = nullptr; variable
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 182 int device_id() const { return device_id_; } in device_id()
233 const int device_id_; variable
|
| D | tpu_client.cc | 271 device_id_(device_buffer->device_id), in PyTpuBuffer()
398 if (dst_device_id == device_id_) { in CopyToDevice()
|
| /external/adhd/cras/src/tests/ |
| D | audio_thread_unittest.cc | 152 iodev->info.idx = ++device_id_; in SetupDevice()
169 device_id_ = 0; in ResetStubData()
233 int device_id_; member in StreamDeviceSuite
|