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 | 121 int conv_width = 1; // Width for the convolution step. in operator() local
123 conv_width *= output->dim_size(i); in operator()
130 output->shaped<T, 2>({conv_width, filter.dim_size(3)}), in operator()
131 input.shaped<T, 2>({conv_width, filter.dim_size(2)}), in operator()
|
| D | conv_ops.cc | 95 int conv_width = 1; // Width for the convolution step. in operator ()() local
97 conv_width *= output->dim_size(i); in operator ()()
104 output->shaped<T, 2>({conv_width, filter.dim_size(3)}), in operator ()()
105 input.shaped<T, 2>({conv_width, filter.dim_size(2)}), in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 5481 int64 conv_width = 1; in SimplifyConvToDot() local
5484 conv_width *= input_shape.dimensions(i); in SimplifyConvToDot()
5491 input_shape.element_type(), {conv_width, input_channels}); in SimplifyConvToDot()
5500 convolution_shape.element_type(), {conv_width, output_channels}); in SimplifyConvToDot()
|