Searched refs:gpu_index (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_desc.h | 388 int gpu_index = s * element_size + c; in DataFromLinear() local
389 dst[gpu_index] = value; in DataFromLinear()
415 int gpu_index = desc.GetLinearIndex(shape, b, x, y, d, s, c); in DataFromBHWDC() local
416 dst[gpu_index] = value; in DataFromBHWDC()
442 int gpu_index = desc.GetLinearIndex(shape, b, x, y, d, s, c); in DataToBHWDC() local
443 dst[cpu_index] = src[gpu_index]; in DataToBHWDC()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.-eager-tensor.pbtxt | 78 argspec: "args=[\'self\', \'gpu_index\'], varargs=None, keywords=None, defaults=[\'0\'], "
|
| /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 | 1184 gpu_index = -1
1193 gpu_index += 1
1200 visible_device_list.append(str(gpu_index))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 1284 def gpu(self, gpu_index=0): argument
1295 return self._copy(context.context(), "GPU:" + str(gpu_index))
|