Searched refs:input_image_size (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | MaxUnpoolKernel.cpp | 54 int64_t input_image_size = numel / channels; in cpu_max_unpool() local
63 data_index_init(begin, c, channels, ip, input_image_size); in cpu_max_unpool()
77 data_index_step(c, channels, ip, input_image_size); in cpu_max_unpool()
118 int64_t input_image_size = input_height * input_width; in cpu_max_unpool_channels_last() local
124 at::parallel_for(0, nbatch * input_image_size, 0, [&](int64_t begin, int64_t end) { in cpu_max_unpool_channels_last()
127 data_index_init(begin, n, nbatch, ip, input_image_size); in cpu_max_unpool_channels_last()
146 data_index_step(n, nbatch, ip, input_image_size); in cpu_max_unpool_channels_last()
191 int64_t input_image_size = numel / channels; in cpu_max_unpool_backward() local
200 data_index_init(begin, c, channels, ip, input_image_size); in cpu_max_unpool_backward()
214 data_index_step(c, channels, ip, input_image_size); in cpu_max_unpool_backward()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | transpose_conv.h | 44 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local
52 const int input_offset = input_image_size * input_depth; in TransposeConvV2()
82 rhs_params.cols = input_image_size; in TransposeConvV2()
88 dst_params.cols = input_image_size; in TransposeConvV2()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 198 const int64_t input_image_size = in operator ()() local
217 const int64_t in_base = b * input_image_size; in operator ()()
|
| D | depthwise_conv_grad_op.cc | 425 const int64_t input_image_size = in operator ()() local
462 in_backprop + b * input_image_size); in operator ()()
914 const int64_t input_image_size = in operator ()() local
931 input + b * input_image_size, input_buffer_data); in operator ()()
|
| D | pooling_ops_common.h | 583 const int64_t input_image_size =
590 in_flat.data() + start * input_image_size, params.depth,
|
| D | deep_conv2d.cc | 1132 const int64_t input_image_size = args.in_rows * args.in_cols * in_depth; in operator ()() local
1140 const int64_t in_base = b * input_image_size; in operator ()()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4965 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local
4973 const int input_offset = input_image_size * input_depth; in TransposeConvV2()
5000 rhs_params.cols = input_image_size; in TransposeConvV2()
5004 dst_params.cols = input_image_size; in TransposeConvV2()
5464 const int input_image_size = input_shape.Dims(1) * input_shape.Dims(2); in TransposeConvV2() local
5472 const int input_offset = input_image_size * input_depth; in TransposeConvV2()
5501 rhs_params.cols = input_image_size; in TransposeConvV2()
5507 dst_params.cols = input_image_size; in TransposeConvV2()
|