Home
last modified time | relevance | path

Searched refs:RunAndFetch (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_fused_ops_test.cc68 static void RunAndFetch(const tensorflow::Scope& root, const string& fetch, in RunAndFetch() function in tensorflow::CommonTestUtilities
147 CommonTestUtilities<T>::RunAndFetch(root, "with_bias", output); in RunConv2DWithBias()
168 CommonTestUtilities<T>::RunAndFetch(root, "with_relu", output); in RunConv2DWithBiasAndRelu()
562 CommonTestUtilities<T>::RunAndFetch(root, "with_bias", output); in RunMklPadWithFusedConv2DAndBias()
592 CommonTestUtilities<T>::RunAndFetch(root, "with_relu", output); in RunMklPadWithFusedConv2DAndBiasRelu()
Dconv_ops_test.cc551 void RunAndFetch(const tensorflow::Scope& root, const string& fetch, in RunAndFetch() function in tensorflow::FusedConv2DOpTest
624 RunAndFetch(root, "with_bias", output, allow_gpu_device); in RunConv2DWithBias()
645 RunAndFetch(root, "with_relu", output, allow_gpu_device); in RunConv2DWithBiasAndRelu()
674 RunAndFetch(root, "with_fused_batch_norm", output, allow_gpu_device); in RunConv2DWithBatchNorm()
704 RunAndFetch(root, "with_relu", output, allow_gpu_device); in RunConv2DWithBatchNormAndRelu()
740 RunAndFetch(root, fused_conv2d.name(), output, allow_gpu_device, in RunFusedConv2DOp()