Searched defs:gpu_index (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| H A D | tensor_desc.h | 388 int gpu_index = s * element_size + c; in DataFromLinear() local
415 int gpu_index = desc.GetLinearIndex(shape, b, x, y, d, s, c); in DataFromBHWDC() local
442 int gpu_index = desc.GetLinearIndex(shape, b, x, y, d, s, c); in DataToBHWDC() local
|
| /aosp_15_r20/external/mesa3d/src/amd/common/ |
| H A D | ac_rgp.c | 307 int32_t gpu_index; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | ops.py | 1284 def gpu(self, gpu_index=0): argument
|