Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc34 tensorflow::int64 lhs_col_dilation, tensorflow::int64 rhs_row_dilation, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
41 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_EigenSingleThreadedConvF16()
56 tensorflow::int64 rhs_row_dilation, tensorflow::int64 rhs_col_dilation) { in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
62 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc37 tensorflow::int64 rhs_row_dilation, tensorflow::int64 rhs_col_dilation) { in __xla_cpu_runtime_EigenConvF32() argument
46 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_EigenConvF32()
60 tensorflow::int64 lhs_col_dilation, tensorflow::int64 rhs_row_dilation, in __xla_cpu_runtime_EigenConvF16() argument
70 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc60 int64 rhs_row_dilation, int64 rhs_col_dilation) { in MKLConvImpl() argument
77 memory::dims conv1_dilates = {ToInt(rhs_row_dilation - 1), in MKLConvImpl()
161 int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) { in __xla_cpu_runtime_MKLConvF32() argument
171 lhs_row_dilation, lhs_col_dilation, rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_MKLConvF32()
178 rhs_row_dilation, rhs_col_dilation); in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d.h35 tensorflow::int64 lhs_col_dilation, tensorflow::int64 rhs_row_dilation,
50 tensorflow::int64 rhs_row_dilation, tensorflow::int64 rhs_col_dilation);
Druntime_single_threaded_conv2d.h36 tensorflow::int64 rhs_row_dilation, tensorflow::int64 rhs_col_dilation);
49 tensorflow::int64 lhs_col_dilation, tensorflow::int64 rhs_row_dilation,
Druntime_conv2d_impl.h42 Eigen::Index lhs_col_dilation, Eigen::Index rhs_row_dilation, in EigenConvImpl() argument
82 row_stride, rhs_col_dilation, rhs_row_dilation, in EigenConvImpl()
Druntime_conv2d_mkl.h35 tensorflow::int64 lhs_col_dilation, tensorflow::int64 rhs_row_dilation,
Dir_emitter.cc950 int64 rhs_row_dilation = window.dimensions(0).window_dilation(); in HandleConvolution() local
1002 b_.getInt64(rhs_row_dilation), in HandleConvolution()