| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_impl.h | 37 Eigen::Index kernel_filters, Eigen::Index output_rows, in EigenConvImpl() argument
50 kernel(rhs, kernel_rows, kernel_cols, kernel_channels, kernel_filters); in EigenConvImpl()
54 output(out, input_batch, output_rows, output_cols, kernel_filters); in EigenConvImpl()
72 post_contract_dims[3] = kernel_filters; in EigenConvImpl()
76 kernel_dims[1] = kernel_filters; in EigenConvImpl()
|
| D | runtime_conv2d_mkl.cc | 55 int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, in MKLConvImpl() argument
70 memory::dims conv1_weights_dim = {ToInt(kernel_filters), in MKLConvImpl()
73 memory::dims conv1_dst_dim = {ToInt(input_batch), ToInt(kernel_filters), in MKLConvImpl()
158 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_MKLConvF32() argument
169 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_MKLConvF32()
175 kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_single_threaded_conv2d.cc | 29 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
39 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
50 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
60 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 31 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()
55 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows, in __xla_cpu_runtime_EigenConvF16() argument
68 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_conv2d.h | 30 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows,
44 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters,
|
| D | runtime_single_threaded_conv2d.h | 30 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters,
44 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows,
|
| D | runtime_conv2d_mkl.h | 30 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows,
|
| D | ir_emitter.cc | 922 int64 kernel_filters = in HandleConvolution() local
991 b_.getInt64(kernel_filters), in HandleConvolution()
|