Home
last modified time | relevance | path

Searched refs:input_depth (Results 1 – 25 of 84) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc31 const int input_depth = 2; in TEST() local
40 Tensor<float, 3> input_backward(input_depth, input_rows, input_cols); in TEST()
41 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
51 EXPECT_EQ(input_backward.dimension(0), input_depth); in TEST()
55 for (int id = 0; id < input_depth; ++id) { in TEST()
80 const int input_depth = 2; in TEST() local
90 input_depth); in TEST()
91 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, in TEST()
105 EXPECT_EQ(input_backward.dimension(2), input_depth); in TEST()
107 for (int id = 0; id < input_depth; ++id) { in TEST()
[all …]
Deigen_backward_cuboid_convolutions_test.cc31 const int input_depth = 2; in TEST() local
43 Tensor<float, 4> input_backward(input_depth, input_planes, input_rows, in TEST()
45 Tensor<float, 5> kernel(output_depth, input_depth, patch_planes, patch_rows, in TEST()
60 EXPECT_EQ(input_backward.dimension(0), input_depth); in TEST()
62 for (int id = 0; id < input_depth; ++id) { in TEST()
94 const int input_depth = 2; in TEST() local
107 input_planes, input_depth); in TEST()
109 input_depth, output_depth); in TEST()
123 EXPECT_EQ(input_backward.dimension(3), input_depth); in TEST()
125 for (int id = 0; id < input_depth; ++id) { in TEST()
[all …]
Dconv_ops_using_gemm.cc92 int input_depth, const T2* filter_data, int filter_height, in operator ()() argument
159 for (int in_channel = 0; in_channel < input_depth; in operator ()()
170 input_depth) + in operator ()()
171 (in_y * input_width * input_depth) + in operator ()()
172 (in_x * input_depth) + in_channel]; in operator ()()
177 filter_data[(filter_y * filter_width * input_depth * in operator ()()
179 (filter_x * input_depth * filter_count) + in operator ()()
215 int input_depth, const T2* filter_data, int filter_height, in operator ()() argument
220 (input_depth <= 0)) { in operator ()()
223 << input_width << ", " << input_depth; in operator ()()
[all …]
Deigen_benchmark_cpu_test.cc33 int input_depth, in SpatialConvolution() argument
43 input_width, input_depth); in SpatialConvolution()
45 input_depth, filter_count); in SpatialConvolution()
50 (input_dims.TotalSize() / input_depth) * filter_count; in SpatialConvolution()
52 num_computed_elements * (input_depth * filter_height * filter_width); in SpatialConvolution()
60 int input_width, int input_depth, in SpatialConvolutionBackwardInput() argument
71 input_width, input_depth); in SpatialConvolutionBackwardInput()
73 input_depth, filter_count); in SpatialConvolutionBackwardInput()
79 num_computed_elements * (input_depth * filter_height * filter_width); in SpatialConvolutionBackwardInput()
87 int input_width, int input_depth, in SpatialConvolutionBackwardKernel() argument
[all …]
Dquantized_batch_norm_op_test.cc65 const int input_depth = 2; in TEST_F() local
67 {input_batch, input_height, input_width, input_depth}); in TEST_F()
74 Tensor mean_float(DT_FLOAT, {input_depth}); in TEST_F()
80 Tensor variance_float(DT_FLOAT, {input_depth}); in TEST_F()
86 Tensor beta_float(DT_FLOAT, {input_depth}); in TEST_F()
92 Tensor gamma_float(DT_FLOAT, {input_depth}); in TEST_F()
121 TensorShape({input_batch, input_height, input_width, input_depth})); in TEST_F()
162 const int input_depth = 2; in TEST_F() local
164 {input_batch, input_height, input_width, input_depth}); in TEST_F()
171 Tensor mean_float(DT_FLOAT, {input_depth}); in TEST_F()
[all …]
Dconv_ops_fused_image_transform.cc112 int input_depth; member
171 int64_t cache_line_width, int64_t input_width, int64_t input_depth, in CalculatePerCacheLineParameters() argument
186 resize_cache + (cache_index_y * cache_line_width * input_depth); in CalculatePerCacheLineParameters()
203 input_batch_start + (top_y_index * input_width * input_depth); in CalculatePerCacheLineParameters()
205 input_batch_start + (bottom_y_index * input_width * input_depth); in CalculatePerCacheLineParameters()
228 T1* cache_line_start, int64_t input_depth, in CalculatePerCachePixelParameters() argument
235 result.cache_line_pixel = cache_line_start + (cache_index_x * input_depth); in CalculatePerCachePixelParameters()
262 int padded_height, int padded_width, int input_depth, in operator ()() argument
270 (input_depth <= 0)) { in operator ()()
273 << padded_width << ", " << input_depth; in operator ()()
[all …]
Dquantized_conv_ops.cc59 int input_depth, int input_offset, const T2* filter_data, in operator ()() argument
137 for (int in_channel = 0; in_channel < input_depth; in operator ()()
148 input_depth) + in operator ()()
149 (in_y * input_width * input_depth) + in operator ()()
150 (in_x * input_depth) + in_channel]; in operator ()()
159 filter_data[(filter_y * filter_width * input_depth * in operator ()()
161 (filter_x * input_depth * filter_count) + in operator ()()
205 int input_depth, int input_offset, const T2* filter_data, in operator ()() argument
226 input_width, input_depth, input_offset, filter_data, in operator ()()
264 const int filter_value_count = filter_width * filter_height * input_depth; in operator ()()
[all …]
Dspacetodepth_op_gpu.cu.cc36 const int input_depth, const int output_height, in S2D_NHWC() argument
41 const int d = inp_idx % input_depth; in S2D_NHWC()
42 const int inp_idx2 = inp_idx / input_depth; in S2D_NHWC()
52 const int offset_d = (offset_h * block_size + offset_w) * input_depth; in S2D_NHWC()
150 const int input_depth = input.dimension(3); in operator ()() local
156 batch_size * input_height * input_width * input_depth; in operator ()()
164 input_height, input_width, input_depth, output_height, output_width, in operator ()()
178 const int input_depth = input.dimension(1); in operator ()() local
189 const int input_depth_by_output_area = input_depth * output_area; in operator ()()
229 input_depth * output_height, output.data())); in operator ()()
Deigen_spatial_convolutions_test.cc30 const int input_depth = 7; in TEST() local
39 Tensor<float, 3> input(input_depth, input_rows, input_cols); in TEST()
40 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols); in TEST()
59 for (int id = 0; id < input_depth; ++id) { in TEST()
75 const int input_depth = 7; in TEST() local
84 Tensor<float, 3, RowMajor> input(input_cols, input_rows, input_depth); in TEST()
85 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, in TEST()
104 for (int id = 0; id < input_depth; ++id) { in TEST()
194 const int input_depth = 10; in TEST() local
204 Tensor<float, 4> input(input_depth, input_rows, input_cols, num_batches); in TEST()
[all …]
Ddepthtospace_op_gpu.cu.cc37 const int input_depth, const int output_height, in D2S_NHWC() argument
56 in_d + input_depth * (in_w + input_width * (in_h + input_height * b)); in D2S_NHWC()
153 const int input_depth = input.dimension(3); in operator ()() local
167 input_height, input_width, input_depth, output_height, output_width, in operator ()()
181 const int input_depth = input.dimension(1); in operator ()() local
186 const int input_depth_by_input_area = input_depth * input_area; in operator ()()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_float.h36 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
97 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
180 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
189 for (; ic <= input_depth - 16; ic += 16) {
220 for (; ic <= input_depth - 4; ic += 4) {
239 for (; ic < input_depth; ic++) {
251 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
260 for (; ic <= input_depth - 2; ic += 2) {
287 for (; ic < input_depth; ic++) {
321 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
[all …]
Dmultithreaded_conv.h89 int input_depth, const T* filter_data, int filter_height, in operator()
104 input_depth); in operator()
105 ConstEigenMatrix filter(filter_data, input_depth, filter_count); in operator()
113 filter_width * filter_height * input_depth; in operator()
125 input_width, input_depth); in operator()
127 input_depth, filter_count); in operator()
161 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv() local
171 input_depth, filter_data, filter_height, filter_width, in Conv()
Dim2col_utils.h142 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in DilatedIm2col() local
153 const RuntimeShape col_shape({1, filter_height, filter_width, input_depth}); in DilatedIm2col()
184 memcpy(dst, src, input_depth * sizeof(T)); in DilatedIm2col()
187 memset(dst, zero_byte, input_depth * sizeof(T)); in DilatedIm2col()
195 memset(dst, zero_byte, filter_width * input_depth * sizeof(T)); in DilatedIm2col()
226 const int input_depth = input_shape.Dims(3); in Im2col() local
240 pad_width, pad_height, input_width, input_height, input_depth, in Im2col()
263 const int input_depth = input_shape.Dims(3); in Im2col() local
278 pad_width, pad_height, input_width, input_height, input_depth, in Im2col()
389 const int input_depth = input_shape.Dims(1); in Im2col3D() local
[all …]
Ddepthwiseconv_uint8_3x3_filter.h101 static_assert(offsetof(DepthwiseConvParams, input_depth) == OFFSET_INPUT_DEPTH,
207 static_assert(offsetof(DepthwiseConvDotProdParams, input_depth) ==
325 int64_t input_depth, int64_t input_row_size,
328 const int64_t input_width_increment = 2 * input_depth;
1222 [input_depth] "r"(input_depth),
1258 int64_t input_depth, int64_t input_row_size,
1261 const int64_t input_width_increment = 2 * input_depth;
2047 [input_depth] "r"(input_depth),
2083 int64_t input_depth, int64_t input_row_size,
2086 const int64_t input_width_increment = 4 * input_depth;
[all …]
Ddepthwiseconv_3x3_filter_common.h259 const int32 input_depth = input_shape.Dims(3);
268 (((input_depth % 8) == 0 && depth_multiplier == 1) ||
269 (input_depth == 1 && depth_multiplier > 1));
309 int64_t input_depth;
336 int64_t input_depth;
399 inline void ShuffleInput(const T* input_ptr, int64_t input_depth,
403 const int64_t input_row_size = input_depth * input_width;
409 ptr += input_depth;
447 const int32 input_depth = input_shape.Dims(3);
459 (input_depth % 8) == 0 && (output_shift <= 0) &&
[all …]
Ddepthwiseconv_uint8.h43 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
90 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
158 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
228 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
305 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
371 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
485 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
545 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
640 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
710 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid_3x3_filter.h60 static_assert(offsetof(DepthwiseConvParams, input_depth) == OFFSET_INPUT_DEPTH,
132 float* output_ptr, int64_t input_depth,
137 const int64_t input_width_increment = 2 * input_depth;
1030 [input_depth] "r"(input_depth),
1066 float* output_ptr, int64_t input_depth,
1071 const int64_t input_width_increment = 4 * input_depth;
2036 [input_depth] "r"(input_depth),
2795 int64_t end_depth, int64_t input_depth, int64_t input_row_size,
2803 input_depth,
2868 ptr += params.input_depth;
[all …]
Ddepthwise_conv_3x3_filter.h58 static_assert(offsetof(DepthwiseConvParams, input_depth) == OFFSET_INPUT_DEPTH,
115 int8* output_ptr, int64_t input_depth,
119 const int64_t input_width_increment = 2 * input_depth;
933 [input_depth] "r"(input_depth),
970 int8* output_ptr, int64_t input_depth,
974 const int64_t input_width_increment = 4 * input_depth;
1894 [input_depth] "r"(input_depth),
2628 int64_t input_depth, int64_t input_row_size, int32 output_window_height,
2636 input_depth,
2701 ptr += params.input_depth;
[all …]
Ddepthwise_conv.h48 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
94 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
161 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
229 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
303 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
366 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
476 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
534 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
626 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
693 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv_test.cc108 int input_depth = GetShape(input_)[3]; in BaseDepthwiseConvolutionOpModel() local
110 int depth_mul = output_depth / input_depth; in BaseDepthwiseConvolutionOpModel()
478 const int input_depth = 4; in BatchPaddingValidTest() local
485 {input_batch, input_height, input_width, input_depth}}, in BatchPaddingValidTest()
539 const int input_depth = 1; in BatchPaddingSameTest() local
546 {input_batch, input_height, input_width, input_depth}}, in BatchPaddingSameTest()
1072 const int input_depth = 8; in TEST_P() local
1080 {input_batch, input_height, input_width, input_depth}, in TEST_P()
1145 const int input_depth = 8; in TEST_P() local
1153 {input_batch, input_height, input_width, input_depth}, in TEST_P()
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_helpers.cc155 int input_depth, in CheckShouldSplitDwConv() argument
163 int weights_depth_size = input_depth * weight_shape_[3]; in CheckShouldSplitDwConv()
171 input_depth > kDwConv5x5Filt2x2StrideChannelCount && in CheckShouldSplitDwConv()
172 input_depth % kDwConv5x5Filt2x2StrideChannelCount == 0 && in CheckShouldSplitDwConv()
183 const std::vector<uint8_t>& converted_data, int input_depth, in SplitWeightsForDwConv() argument
212 input_depth, channel_multiplier}; in SplitWeightsForDwConv()
226 TfLiteContext* context, const int input_depth) { in InitializeWeightsNodes() argument
292 const int channel_multiplier = weights_depth_size / input_depth; in InitializeWeightsNodes()
293 weight_shape_ = {weights_height_size, weights_width_size, input_depth, in InitializeWeightsNodes()
298 CheckShouldSplitDwConv(weights_tensor.type, input_depth, in InitializeWeightsNodes()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h55 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
60 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
66 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel()
152 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
157 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
163 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel()
235 const int input_depth = input_shape.Dims(3); in DepthwiseConvHybridPerChannel() local
241 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvHybridPerChannel()
247 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvHybridPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h48 const int input_depth = input_shape.Dims(3); in Conv() local
58 const int groups = input_depth / filter_input_depth; in Conv()
59 TFLITE_DCHECK_EQ(input_depth % filter_input_depth, 0); in Conv()
138 const int input_depth = input_shape.Dims(3); in Conv() local
148 const int groups = input_depth / filter_input_depth; in Conv()
149 TFLITE_DCHECK_EQ(input_depth % filter_input_depth, 0); in Conv()
225 const int input_depth = input_shape.Dims(3); in HybridConvPerChannel() local
235 const int groups = input_depth / filter_input_depth; in HybridConvPerChannel()
236 TFLITE_DCHECK_EQ(input_depth % filter_input_depth, 0); in HybridConvPerChannel()
Dspace_to_depth.h38 const int input_depth = input_shape.Dims(3); in SpaceToDepth() local
52 TFLITE_DCHECK_EQ(input_depth * block_size * block_size, output_depth); in SpaceToDepth()
58 for (int in_d = 0; in_d < input_depth; ++in_d) { in SpaceToDepth()
61 input_depth; in SpaceToDepth()
/external/tensorflow/tensorflow/lite/kernels/internal/
Daveragepool_quantized_test.cc66 const int input_depth = UniformRandomInt(1, 700); in CreateDataAndRunAveragePool() local
67 const int output_depth = input_depth; in CreateDataAndRunAveragePool()
85 RuntimeShape({batch, input_height, input_width, input_depth}); in CreateDataAndRunAveragePool()
129 const int input_depth = UniformRandomInt(1, 700); in CreateExtremalDataAndRunAveragePool() local
130 const int output_depth = input_depth; in CreateExtremalDataAndRunAveragePool()
151 RuntimeShape({batch, input_height, input_width, input_depth}); in CreateExtremalDataAndRunAveragePool()

1234