| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_impl.h | 32 int64 kernel_filters, int64 output_rows, int64 output_cols, in EigenConvImpl() argument
44 kernel(rhs, kernel_rows, kernel_cols, kernel_channels, kernel_filters); in EigenConvImpl()
48 output(out, input_batch, output_rows, output_cols, kernel_filters); in EigenConvImpl()
66 post_contract_dims[3] = kernel_filters; in EigenConvImpl()
70 kernel_dims[1] = kernel_filters; in EigenConvImpl()
|
| D | runtime_single_threaded_conv2d.cc | 29 int64 kernel_channels, int64 kernel_filters, int64 output_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
37 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
47 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
54 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| 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()
156 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_MKLConvF32() argument
167 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_MKLConvF32()
173 kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d.cc | 31 int64 kernel_filters, int64 output_rows, int64 output_cols, 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()
|
| 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.h | 30 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows,
44 tensorflow::int64 kernel_channels, tensorflow::int64 kernel_filters,
|
| D | runtime_conv2d_mkl.h | 30 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows,
|
| D | ir_emitter.cc | 1145 int64 kernel_filters = in HandleConvolution() local
1228 b_.getInt64(kernel_filters), in HandleConvolution()
|