/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
runtime_single_threaded_conv2d.cc | 28 int64 input_channels, int64 kernel_rows, int64 kernel_cols, 36 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, local 46 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, 53 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, local
|
ir_emitter.cc | 893 int64 kernel_cols = local [all...] |
/external/tensorflow/tensorflow/core/framework/ |
common_shape_fns.cc | 663 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); local 686 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); 815 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); local 917 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); local 974 int32 kernel_planes, kernel_rows, kernel_cols; local [all...] |