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 | 29 int64 kernel_channels, int64 kernel_filters, int64 output_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
46 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 30 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_EigenConvF32()
49 int64 kernel_channels, int64 kernel_filters, int64 output_rows, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_conv2d_mkl.cc | 55 int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, in MKLConvImpl()
155 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d_impl.h | 31 int64 kernel_cols, int64 kernel_channels, in EigenConvImpl()
|
| D | ir_emitter.cc | 1143 int64 kernel_channels = in HandleConvolution() local
|