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