Home
last modified time | relevance | path

Searched refs:lhs_col_dilation (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc32 int64 lhs_row_dilation, int64 lhs_col_dilation, int64 rhs_row_dilation, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
39 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_EigenSingleThreadedConvF16()
50 int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) { in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
56 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc34 int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) { in __xla_cpu_runtime_EigenConvF32() argument
42 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_EigenConvF32()
52 int64 lhs_row_dilation, int64 lhs_col_dilation, int64 rhs_row_dilation, in __xla_cpu_runtime_EigenConvF16() argument
61 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc59 int64 lhs_row_dilation, int64 lhs_col_dilation, in MKLConvImpl() argument
159 int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) { in __xla_cpu_runtime_MKLConvF32() argument
163 if (lhs_row_dilation > 1 || lhs_col_dilation > 1) { in __xla_cpu_runtime_MKLConvF32()
169 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_MKLConvF32()
175 padding_right, lhs_row_dilation, lhs_col_dilation, in __xla_cpu_runtime_MKLConvF32()
Druntime_single_threaded_conv2d.h35 tensorflow::int64 lhs_row_dilation, tensorflow::int64 lhs_col_dilation,
49 tensorflow::int64 lhs_col_dilation, tensorflow::int64 rhs_row_dilation,
Druntime_conv2d.h35 tensorflow::int64 lhs_col_dilation, tensorflow::int64 rhs_row_dilation,
49 tensorflow::int64 lhs_row_dilation, tensorflow::int64 lhs_col_dilation,
Druntime_conv2d_impl.h36 int64 lhs_col_dilation, int64 rhs_row_dilation, in EigenConvImpl() argument
77 lhs_col_dilation, lhs_row_dilation, in EigenConvImpl()
Druntime_conv2d_mkl.h35 tensorflow::int64 lhs_col_dilation, tensorflow::int64 rhs_row_dilation,
Dir_emitter.cc1171 int64 lhs_col_dilation = in HandleConvolution() local
1238 b_.getInt64(lhs_col_dilation), in HandleConvolution()