| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d_gpu_int_spatial_convolution.cu.cc | 32 template struct SpatialConvolution<Eigen::GpuDevice, int32>; variable
|
| D | conv_2d.h | 40 output.device(d) = Eigen::SpatialConvolution(
48 struct SpatialConvolution { struct
73 struct SpatialConvolution<Device, Eigen::half, OutputKernel> { argument
82 Eigen::SpatialConvolution(input.cast<float>(), filter.cast<float>(),
96 Eigen::SpatialConvolution(
|
| D | eigen_benchmark_cpu_test.cc | 30 void SpatialConvolution(::testing::benchmark::State& state, int num_threads, in SpatialConvolution() function
47 benchmark.SpatialConvolution(input_dims, filter_dims); in SpatialConvolution()
124 static void BM_SPATIAL_NAME(SpatialConvolution, NT, N, H, W, C, FC, FH, \
127 SpatialConvolution(state, NT, N, H, W, C, FC, FH, FW); \
129 BENCHMARK(BM_SPATIAL_NAME(SpatialConvolution, NT, N, H, W, C, FC, FH, FW))
|
| 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 | 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
1166 SpatialConvolution, //
1176 SpatialConvolution, //
1186 SpatialConvolution, //
1194 SpatialConvolution, //
|
| D | eigen_benchmark.h | 46 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 | 125 functor::SpatialConvolution<Device, T>()( in operator ()()
133 functor::SpatialConvolution<Device, T>()( in operator ()()
1137 void SpatialConvolution<GPUDevice, T>::operator()( \
1145 void SpatialConvolution<GPUDevice, T>::operator()( \
1152 extern template struct SpatialConvolution<GPUDevice, T>; \
|
| D | conv_ops_fused_impl.h | 154 functor::SpatialConvolution<CPUDevice, T, OutputKernelWrapper>()( in operator()
164 functor::SpatialConvolution<CPUDevice, T, OutputKernelWrapper>()( in operator()
|
| D | eigen_spatial_convolutions-inl.h | 1599 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/core/kernels/mkl/ |
| D | mkl_fused_ops_test.cc | 392 TYPED_TEST_P(MklFusedConv2DWithBiasOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
529 MklFusedConv2DWithBiasOpTest, OneByOneConvolution, SpatialConvolution,
713 TYPED_TEST_P(MklFusedDepthwiseConv2DWithBiasOpTest, SpatialConvolution) { in TYPED_TEST_P() argument
774 SpatialConvolution, OneByOneConvolutionAndRelu, SpatialConvolutionAndRelu,
|