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_single_threaded_conv2d.cc29 int64 kernel_channels, int64 kernel_filters, int64 output_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
36 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
46 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
53 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc30 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_EigenConvF32() argument
40 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF32()
49 int64 kernel_channels, int64 kernel_filters, int64 output_rows, in __xla_cpu_runtime_EigenConvF16() argument
59 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_impl.h31 int64 kernel_cols, int64 kernel_channels, in EigenConvImpl() argument
44 kernel(rhs, kernel_rows, kernel_cols, kernel_channels, kernel_filters); in EigenConvImpl()
59 pre_contract_dims[1] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
69 kernel_dims[0] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
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()
155 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_MKLConvF32() argument
166 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConvF32()
172 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConvF32()
Druntime_single_threaded_conv2d.h30 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters,
43 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
Druntime_conv2d.h29 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
44 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters,
Druntime_conv2d_mkl.h29 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
Dir_emitter.cc1143 int64 kernel_channels = in HandleConvolution() local
1227 b_.getInt64(kernel_channels), in HandleConvolution()