Searched refs:rhs_col_dilation (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_conv2d.cc | 33 int64 rhs_col_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()
|
| D | runtime_conv2d.cc | 34 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()
53 int64 rhs_col_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()
|
| D | runtime_conv2d_mkl.cc | 60 int64 rhs_row_dilation, int64 rhs_col_dilation) { in MKLConvImpl() argument
78 ToInt(rhs_col_dilation - 1)}; in MKLConvImpl()
159 int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) { in __xla_cpu_runtime_MKLConvF32() argument
169 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_MKLConvF32()
176 rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_single_threaded_conv2d.h | 36 tensorflow::int64 rhs_row_dilation, tensorflow::int64 rhs_col_dilation);
50 tensorflow::int64 rhs_col_dilation);
|
| D | runtime_conv2d.h | 36 tensorflow::int64 rhs_col_dilation);
50 tensorflow::int64 rhs_row_dilation, tensorflow::int64 rhs_col_dilation);
|
| D | runtime_conv2d_impl.h | 37 int64 rhs_col_dilation) { in EigenConvImpl() argument
76 row_stride, rhs_col_dilation, rhs_row_dilation, in EigenConvImpl()
|
| D | runtime_conv2d_mkl.h | 36 tensorflow::int64 rhs_col_dilation);
|
| D | ir_emitter.cc | 1174 int64 rhs_col_dilation = in HandleConvolution() local
1240 b_.getInt64(rhs_col_dilation), in HandleConvolution()
|