Searched refs:gpu_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.cc | 592 const int gpu_index = GetLinearIndex(b, x, y, d, s, c); in DataFromBHWDC() local
593 dst[gpu_index] = value; in DataFromBHWDC()
620 const int gpu_index = GetLinearIndex(b, x, y, d, s, c); in DataToBHWDC() local
621 dst[cpu_index] = src[gpu_index]; in DataToBHWDC()
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_rgp.c | 253 int32_t gpu_index; member
377 chunk->gpu_index = 0; in radv_fill_sqtt_asic_info()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 949 gpu_index = -1
952 gpu_index += 1
959 visible_device_list.append(str(gpu_index))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 1100 def gpu(self, gpu_index=0): argument
1111 return self._copy(context.context(), "GPU:" + str(gpu_index))
|