Searched defs:SpatialConvolution (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_2d.h | 61 struct SpatialConvolution { struct
68 SpatialConvolutionFunc(d, output, input, filter, row_stride, col_stride, in operator() argument
77 SpatialConvolutionFunc(d, output, input, filter, row_stride, col_stride, in operator() argument
88 SpatialConvolutionFunc( in operator() argument
108 struct SpatialConvolution<Device, Eigen::half, OutputKernel> { argument
|
| H A D | conv_2d_gpu_int_spatial_convolution.cu.cc | 32 template struct SpatialConvolution<Eigen::GpuDevice, int32>; variable
|
| H A D | eigen_benchmark.h | 46 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution() function
|
| H A D | conv_ops_test.cc | 975 TYPED_TEST_P(FusedConv2DWithBiasOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
1048 TYPED_TEST_P(FusedConv2DWithBatchNormOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
|
| H A D | eigen_benchmark_cpu_test.cc | 30 void SpatialConvolution(::testing::benchmark::State& state, int num_threads, in SpatialConvolution() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_fused_ops_test.cc | 372 TYPED_TEST_P(MklFusedConv2DWithBiasOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
661 TYPED_TEST_P(MklFusedDepthwiseConv2DWithBiasOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
|