Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d_impl.h36 Eigen::Index kernel_cols, Eigen::Index kernel_channels, in EigenConvImpl() argument
50 kernel(rhs, kernel_rows, kernel_cols, kernel_channels, kernel_filters); in EigenConvImpl()
65 pre_contract_dims[1] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
75 kernel_dims[0] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
Druntime_single_threaded_conv2d.cc28 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
38 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
50 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
59 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc31 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters, in __xla_cpu_runtime_EigenConvF32() argument
44 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF32()
54 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels, in __xla_cpu_runtime_EigenConvF16() argument
68 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc55 int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, in MKLConvImpl() argument
71 ToInt(kernel_channels), ToInt(kernel_rows), in MKLConvImpl()
157 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_MKLConvF32() argument
168 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConvF32()
174 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d.h29 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
44 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters,
Druntime_single_threaded_conv2d.h30 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters,
43 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
Druntime_conv2d_mkl.h29 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
Dir_emitter.cc920 int64 kernel_channels = in HandleConvolution() local
990 b_.getInt64(kernel_channels), in HandleConvolution()