Searched defs:kernel_channels (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_conv2d.cc | 28 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
50 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 31 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters, in __xla_cpu_runtime_EigenConvF32()
54 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_conv2d_mkl.cc | 55 int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, in MKLConvImpl()
157 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d_impl.h | 36 Eigen::Index kernel_cols, Eigen::Index kernel_channels, in EigenConvImpl()
|
| D | ir_emitter.cc | 920 int64 kernel_channels = in HandleConvolution() local
|