Searched defs:allow_gpu_device (Results 1 – 1 of 1) sorted by relevance
552 Tensor* output, bool allow_gpu_device, in RunAndFetch()611 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBias()630 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBiasAndRelu()654 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBatchNorm()681 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBatchNormAndRelu()710 bool allow_gpu_device = false, int stride = 1) { in RunFusedConv2DOp()