Searched defs:gpu_index (Results 1 – 3 of 3) 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
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
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_rgp.c | 253 int32_t gpu_index; member
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 1284 def gpu(self, gpu_index=0): argument
|