| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_benchmark_cpu_test.cc | 30 void SpatialConvolution(int iters, int num_threads, in SpatialConvolution() function
49 benchmark.SpatialConvolution(input_dims, filter_dims); in SpatialConvolution()
128 static void BM_SPATIAL_NAME(SpatialConvolution, NT, N, H, W, C, FC, FH, \
131 SpatialConvolution(iters, NT, N, H, W, C, FC, FH, FW); \
133 BENCHMARK(BM_SPATIAL_NAME(SpatialConvolution, NT, N, H, W, C, FC, FH, FW))
|
| D | conv_2d.h | 63 Eigen::SpatialConvolution(input, filter, col_stride, row_stride, padding, in SpatialConvolutionFunc()
69 struct SpatialConvolution { struct
82 struct SpatialConvolution<Device, Eigen::half, OutputKernel> { argument
91 Eigen::SpatialConvolution(input.cast<float>(), filter.cast<float>(),
|
| D | eigen_spatial_convolutions_test.cc | 47 result = SpatialConvolution(input, kernel); in TEST()
92 result = SpatialConvolution(input, kernel); in TEST()
127 result = SpatialConvolution(input, kernel); in TEST()
164 result = SpatialConvolution(input, kernel); in TEST()
214 result = SpatialConvolution(input, kernel, stride, stride, PADDING_VALID); in TEST()
269 SpatialConvolution(input, kernel, row_stride, col_stride, PADDING_VALID); in TEST()
328 result = SpatialConvolution(input, kernel, stride, stride, PADDING_VALID); in TEST()
380 result = SpatialConvolution(input, kernel, stride, stride, PADDING_VALID); in TEST()
428 result = SpatialConvolution(input, kernel, stride, stride, PADDING_VALID); in TEST()
479 result = SpatialConvolution(input, kernel, stride, stride, PADDING_VALID); in TEST()
[all …]
|
| D | eigen_benchmark.h | 45 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution() function
67 output.device(device_) = Eigen::SpatialConvolution(input, filter); in SpatialConvolution()
|
| D | conv_ops.cc | 113 functor::SpatialConvolution<Device, T>()( in operator ()()
922 void SpatialConvolution<GPUDevice, T>::operator()( \
929 extern template struct SpatialConvolution<GPUDevice, T>; \
|
| D | mkl_fused_ops_test.cc | 280 TYPED_TEST_P(MklFusedConv2DWithBiasOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
300 SpatialConvolution, //
|
| D | conv_ops_test.cc | 974 TYPED_TEST_P(FusedConv2DWithBiasOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
1014 TYPED_TEST_P(FusedConv2DWithBatchNormOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
1041 SpatialConvolution, //
1049 SpatialConvolution, //
|
| D | conv_ops_fused_impl.h | 294 functor::SpatialConvolution<CPUDevice, T, OutputKernel>()( in operator()
|
| D | eigen_spatial_convolutions-inl.h | 1358 SpatialConvolution(const Input& input, const Kernel& kernel,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 129 Eigen::SpatialConvolution(input, filter, stride_cols, stride_rows, in operator()
|
| D | eigen_spatial_convolutions.h | 94 SpatialConvolution(const Input& input, const Kernel& kernel,
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 97 SpatialConvolution(const Input& input, const Kernel& kernel,
|