Searched refs:conv_width (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 99 const int conv_width = output_height * output_width; in operator() local
102 EigenMatrix output(output_data, input_batches * conv_width, filter_count); in operator()
103 ConstEigenMatrix input(input_data, input_batches * conv_width, in operator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_fused_impl.h | 262 int conv_width = 1; // Width for the convolution step. in operator() local
264 conv_width *= output->dim_size(i); in operator()
271 output->shaped<T, 2>({conv_width, filter.dim_size(3)}), in operator()
272 input.shaped<T, 2>({conv_width, filter.dim_size(2)}), in operator()
|
| D | conv_ops.cc | 84 int conv_width = 1; // Width for the convolution step. in operator ()() local
86 conv_width *= output->dim_size(i); in operator ()()
93 output->shaped<T, 2>({conv_width, filter.dim_size(3)}), in operator ()()
94 input.shaped<T, 2>({conv_width, filter.dim_size(2)}), in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3608 int64 conv_width = 1; in SimplifyConvToDot() local
3611 conv_width *= input_shape.dimensions(i); in SimplifyConvToDot()
3618 input_shape.element_type(), {conv_width, input_channels}); in SimplifyConvToDot()
3625 convolution_shape.element_type(), {conv_width, output_channels}); in SimplifyConvToDot()
|