Searched refs:run_fused (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_fused_ops_test.cc | 96 const BiasAddGraphRunner& run_fused) { in VerifyBiasAddTensorsClose() argument
113 run_fused(image, filter, bias, &fused_conv_2d); in VerifyBiasAddTensorsClose()
227 const BiasAddGraphRunner run_fused = in VerifyConv2DWithBias() local
236 filter_count, run_default, run_fused); in VerifyConv2DWithBias()
252 const BiasAddGraphRunner run_fused = in VerifyConv2DWithBiasAndRelu() local
261 filter_count, run_default, run_fused); in VerifyConv2DWithBiasAndRelu()
500 const BiasAddGraphRunner run_fused = in VerifyPadAndConv2DWithBias() local
509 filter_count, run_default, run_fused); in VerifyPadAndConv2DWithBias()
525 const BiasAddGraphRunner run_fused = in VerifyPadAndConv2DWithBiasRelu() local
534 filter_count, run_default, run_fused); in VerifyPadAndConv2DWithBiasRelu()
|
| D | conv_ops_test.cc | 748 const BiasAddGraphRunner& run_fused) { in VerifyBiasAddTensorsNear() argument
768 run_fused(image, filter, bias, &fused_conv_2d); in VerifyBiasAddTensorsNear()
788 const BatchNormGraphRunner& run_fused) { in VerifyFusedBatchNormTensorsNear() argument
818 run_fused(image, filter, scale, offset, mean, variance, &fused_conv_2d); in VerifyFusedBatchNormTensorsNear()
846 const BiasAddGraphRunner run_fused = [this](const Tensor& input_data, in VerifyConv2DWithBias() local
855 run_default, run_fused); in VerifyConv2DWithBias()
872 const BiasAddGraphRunner run_fused = in VerifyConv2DWithBiasAndRelu() local
881 run_default, run_fused); in VerifyConv2DWithBiasAndRelu()
900 const BatchNormGraphRunner run_fused = in VerifyConv2DWithBatchNorm() local
912 filter_count, run_default, run_fused); in VerifyConv2DWithBatchNorm()
[all …]
|