Searched refs:output_kernel (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d.h | 60 const OutputKernel& output_kernel) { in SpatialConvolutionFunc() argument
64 col_dilation, row_dilation, output_kernel); in SpatialConvolutionFunc()
75 const OutputKernel& output_kernel = OutputKernel()) { in operator()
77 row_dilation, col_dilation, padding, output_kernel); in operator()
89 const OutputKernel& output_kernel = OutputKernel()) {
93 row_dilation, output_kernel)
142 const OutputKernel& output_kernel = OutputKernel()) {
143 out.device(d) = in0.contract(in1, dim_pair, output_kernel);
|
| D | conv_ops_fused_impl.h | 258 void operator()(const OutputKernel& output_kernel, OpKernelContext* ctx, in operator()
274 dim_pair, output_kernel); in operator()
291 output_kernel); in operator()
298 output_kernel); in operator()
|
| D | conv_ops.cc | 928 const Eigen::NoOpOutputKernel& output_kernel); \
936 const Eigen::NoOpOutputKernel& output_kernel); \
|
| D | eigen_spatial_convolutions-inl.h | 1363 const OutputKernel& output_kernel = OutputKernel()) {
1484 contract_dims, output_kernel)
1490 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
|