Home
last modified time | relevance | path

Searched defs:ConvolutionTest (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconvolution_test_1d.cc43 class ConvolutionTest : public ClientLibraryTestBase { class
203 XLA_TEST_F(ConvolutionTest, Convolve1D_1x2x5_1x2x2_Valid) { in XLA_TEST_F() argument
269 XLA_TEST_F(ConvolutionTest, Convolve1D_1x2x5_1x2x2_WithLHSDilation) { in XLA_TEST_F() argument
300 XLA_TEST_F(ConvolutionTest, Convolve1D_1x2x5_1x2x2_WithLHSAndRHSDilation) { in XLA_TEST_F() argument
H A Dconvolution_test.cc40 class ConvolutionTest : public ClientLibraryTestBase { class
258 XLA_TEST_F(ConvolutionTest, Convolve3D_1x4x2x3x3_2x2x2x3x3_Valid) { in XLA_TEST_F() argument
1525 XLA_TEST_F(ConvolutionTest, Convolve_bf16_1x1x1x2_1x1x1x2_Valid) { in XLA_TEST_F() argument
1550 XLA_TEST_F(ConvolutionTest, DISABLED_ON_GPU_ROCM(NoCudnnAlgorithmPicker)) { in XLA_TEST_F() argument
1570 XLA_TEST_F(ConvolutionTest, ConvolveF32BackwardInputGroupedConvolution) { in XLA_TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dcudnn_deterministic_ops_test.py21 ConvolutionTest = cudnn_deterministic_base.ConvolutionTest variable
H A Dcudnn_d9m_test.py22 ConvolutionTest = cudnn_deterministic_base.ConvolutionTest variable
H A Dcudnn_deterministic_base.py68 class ConvolutionTest(test.TestCase): class
/aosp_15_r20/external/armnn/delegate/test/
H A DConvolutionTestHelper.hpp197 void ConvolutionTest(tflite::BuiltinOperator convolutionOperatorCode, in ConvolutionTest() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_inference_test.cc443 TEST_F(DynamicDimensionInferenceTest, ConvolutionTest) { in TEST_F() argument
H A Ddynamic_padder_test.cc330 TEST_F(DynamicPadderTest, ConvolutionTest) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_test.py1742 class ConvolutionTest(test_lib.TestCase): class