Searched refs:gpu_index (Results 1 – 8 of 8) 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/libopus/dnn/torch/neural-pitch/ |
| D | experiments.py | 23 os.environ["CUDA_VISIBLE_DEVICES"] = str(args.gpu_index)
|
| D | run_crepe.py | 25 os.environ["CUDA_VISIBLE_DEVICES"] = str(args.gpu_index)
|
| D | data_augmentation.py | 34 os.environ["CUDA_VISIBLE_DEVICES"] = str(args.gpu_index)
|
| /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/common/ |
| D | ac_rgp.c | 307 int32_t gpu_index; member
446 chunk->gpu_index = 0; in ac_sqtt_fill_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))
|