Searched refs:allow_gpu_device (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 552 Tensor* output, bool allow_gpu_device, in RunAndFetch() argument
594 const bool place_all_on_gpu = allow_gpu_device && has_gpu_device; in RunAndFetch()
611 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBias() argument
624 RunAndFetch(root, "with_bias", output, allow_gpu_device); in RunConv2DWithBias()
630 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBiasAndRelu() argument
645 RunAndFetch(root, "with_relu", output, allow_gpu_device); in RunConv2DWithBiasAndRelu()
654 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBatchNorm() argument
674 RunAndFetch(root, "with_fused_batch_norm", output, allow_gpu_device); in RunConv2DWithBatchNorm()
681 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBatchNormAndRelu() argument
704 RunAndFetch(root, "with_relu", output, allow_gpu_device); in RunConv2DWithBatchNormAndRelu()
[all …]
|