Searched defs:allow_gpu_device (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | matmul_op_test.cc | 44 Tensor* output, bool allow_gpu_device, in RunAndFetch()
103 bool allow_gpu_device = false) { in RunMatMulWithBias()
122 Tensor* output, bool allow_gpu_device = false) { in RunMatMulWithBiasAndActivation()
154 bool allow_gpu_device = false) { in RunFusedMatMulOp()
|
| H A D | conv_ops_test.cc | 498 Tensor* output, bool allow_gpu_device, in RunAndFetch()
558 Tensor* output, bool allow_gpu_device = false, in RunConv2DWithBias()
580 Tensor* output, bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBiasAndActivation()
614 Tensor* output, bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBatchNorm()
644 bool allow_gpu_device = false, int stride = 1) { in RunConv2DWithBatchNormAndActivation()
687 Tensor* output, bool allow_gpu_device = false, in RunFusedConv2DOp()
|
| H A D | fused_batch_norm_ex_op_test.cc | 83 std::vector<Tensor>* outputs, bool allow_gpu_device, in RunAndFetch()
|