Searched defs:input_depth (Results 1 - 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Ddepthwise_conv_test.cc57 int input_depth = GetShape(input_)[3]; local
59 int depth_mul = output_depth / input_depth;
H A Dconv.cc211 // [filter_count, filter_height, filter_width, input_depth], but for the float
212 // implementation we need them as [filter_height, filter_width, input_depth,
237 int input_depth = input->dims->data[3]; local
241 im2col_size->data[3] = input_depth * filter_height * filter_width;
259 int input_depth = input->dims->data[3]; local
260 hwcn_weights_size->data[0] = (filter_height * filter_width * input_depth);
/external/eigen/unsupported/test/
H A Dcxx11_tensor_image_patch.cpp185 int input_depth = 3; local
191 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
199 VERIFY_IS_EQUAL(result.dimension(0), input_depth); // depth
228 for (int d = 0; d < input_depth; ++d) { // depth
261 int input_depth = 1; local
268 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
272 VERIFY_IS_EQUAL(result.dimension(0), input_depth); // depth
301 for (int d = 0; d < input_depth; ++d) { // depth
334 int input_depth = 3; local
341 Tensor<float, 4> tensor(input_depth, input_row
[all...]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
H A Ddepthwiseconv_float.h37 const int input_depth = ArraySize(input_dims, 0); local
42 TFLITE_DCHECK(output_depth == input_depth * depth_multiplier);
47 for (int ic = 0; ic < input_depth; ++ic) {
H A Ddepthwiseconv_uint8.h43 const int input_depth = ArraySize(input_dims, 0); local
48 TFLITE_DCHECK(output_depth == input_depth * depth_multiplier);
53 for (int ic = 0; ic < input_depth; ++ic) {
H A Dreference_ops.h168 const int input_depth = MatchingArraySize(input_dims, 0, filter_dims, 0); local
188 for (int in_channel = 0; in_channel < input_depth; ++in_channel) {
264 const int input_depth = MatchingArraySize(input_dims, 0, filter_dims, 0); local
282 for (int in_channel = 0; in_channel < input_depth; ++in_channel) {
367 const int input_depth = ArraySize(input_dims, 0); local
379 TFLITE_DCHECK_EQ(input_depth, output_depth * block_size * block_size);
408 const int input_depth = ArraySize(input_dims, 0); local
420 TFLITE_DCHECK_EQ(input_depth * block_size * block_size, output_depth);
426 for (int in_d = 0; in_d < input_depth; ++in_d) {
429 input_depth;
[all...]
/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_batch_norm_op_test.cc66 const int input_depth = 2; local
68 {input_batch, input_height, input_width, input_depth});
75 Tensor mean_float(DT_FLOAT, {input_depth});
81 Tensor variance_float(DT_FLOAT, {input_depth});
87 Tensor beta_float(DT_FLOAT, {input_depth});
93 Tensor gamma_float(DT_FLOAT, {input_depth});
122 TensorShape({input_batch, input_height, input_width, input_depth}));
163 const int input_depth = 2; local
165 {input_batch, input_height, input_width, input_depth});
172 Tensor mean_float(DT_FLOAT, {input_depth});
[all...]
H A Dconv_ops_test.cc162 int input_depth, int resize_width,
172 TensorShape({1, input_height, input_width, input_depth}));
178 input_depth, filter_count}));
218 int input_depth, int y_padding,
226 TensorShape({1, input_height, input_width, input_depth}));
232 input_depth, filter_count}));
161 CompareFusedAndSeparate(int input_width, int input_height, int input_depth, int resize_width, int resize_height, int y_padding, int x_padding, int filter_size, int filter_count, bool resize_align_corners, const string& pad_mode, int stride, const string& padding) argument
217 CompareFusedPadOnlyAndSeparate(int input_width, int input_height, int input_depth, int y_padding, int x_padding, int filter_size, int filter_count, const string& pad_mode, int stride, const string& padding) argument
H A Ddepthtospace_op.cc88 const int input_depth = variable
96 context, input_depth % block_size_sq == 0,
97 errors::InvalidArgument("Input depth dimension ", input_depth,
100 const int output_depth = input_depth / block_size_sq;
H A Ddepthtospace_op_gpu.cu.cc38 const int input_depth, const int output_height,
57 in_d + input_depth * (in_w + input_width * (in_h + input_height * b));
154 const int input_depth = input.dimension(3); local
164 input_height, input_width, input_depth, output_height, output_width,
178 const int input_depth = input.dimension(1); local
183 const int input_depth_by_input_area = input_depth * input_area;
34 D2S_NHWC(const int32 nthreads, const dtype* __restrict__ input_ptr, const int block_size, const int batch_size, const int input_height, const int input_width, const int input_depth, const int output_height, const int output_width, const int output_depth, dtype* __restrict__ output_ptr) argument
H A Deigen_backward_spatial_convolutions_test.cc32 const int input_depth = 2; local
41 Tensor<float, 3> input_backward(input_depth, input_rows, input_cols);
42 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols);
52 EXPECT_EQ(input_backward.dimension(0), input_depth);
56 for (int id = 0; id < input_depth; ++id) {
81 const int input_depth = 2; local
91 input_depth);
92 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth,
106 EXPECT_EQ(input_backward.dimension(2), input_depth);
108 for (int id = 0; id < input_depth;
133 const int input_depth = 2; local
196 const int input_depth = 2; local
259 const int input_depth = 2; local
307 const int input_depth = 2; local
358 const int input_depth = 2; local
429 const int input_depth = 2; local
501 const int input_depth = 2; local
556 const int input_depth = 2; local
612 const int input_depth = 2; local
679 const int input_depth = 2; local
743 test_batched_strided_spatial_convolution_backward_input_valid( const int num_batches, const int input_depth, const int input_rows, const int input_cols, const int output_depth) argument
802 int input_depth = 1; local
819 test_batched_strided_spatial_convolution_backward_input_valid_row_major( const int num_batches, const int input_depth, const int input_rows, const int input_cols, const int output_depth) argument
879 int input_depth = 1; local
897 const int input_depth = 2; local
947 const int input_depth = 2; local
1003 const int input_depth = 2; local
1050 const int input_depth = 2; local
1100 const int input_depth = 2; local
1148 const int input_depth = 2; local
1195 const int input_depth = 2; local
1239 const int input_depth = 2; local
1302 const int input_depth = 2; local
1367 const int input_depth = 2; local
1422 const int input_depth = 2; local
1478 const int input_depth = 2; local
1541 const int input_depth = 2; local
1608 const int input_depth = 2; local
1675 const int input_depth = 2; local
1735 const int input_depth = 2; local
1796 const int input_depth = 2; local
1875 const int input_depth = 2; local
1954 const int input_depth = 2; local
2031 const int input_depth = 2; local
[all...]
H A Deigen_spatial_convolutions_test.cc28 const int input_depth = 7; local
37 Tensor<float, 3> input(input_depth, input_rows, input_cols);
38 Tensor<float, 4> kernel(output_depth, input_depth, patch_rows, patch_cols);
57 for (int id = 0; id < input_depth; ++id) {
73 const int input_depth = 7; local
82 Tensor<float, 3, RowMajor> input(input_cols, input_rows, input_depth);
83 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth,
102 for (int id = 0; id < input_depth; ++id) {
192 const int input_depth = 10; local
202 Tensor<float, 4> input(input_depth, input_row
244 const int input_depth = 10; local
302 const int input_depth = 10; local
358 const int input_depth = 10; local
406 const int input_depth = 2; local
454 const int input_depth = 10; local
505 const int input_depth = 10; local
556 const int input_depth = 10; local
610 const int input_depth = 10; local
[all...]
H A Dspacetodepth_op.cc88 const int input_depth = variable
101 const int output_depth = input_depth * block_size_ * block_size_;
157 const int input_depth = input.dimension(3); local
166 const int offset_d = (offset_h * block_size + offset_w) * input_depth;
167 for (int d = 0; d < input_depth; ++d) {
H A Dspacetodepth_op_gpu.cu.cc36 const int input_depth, const int output_height,
40 // inp_idx = d + input_depth * (w + input_width * (h + input_height * b))
41 const int d = inp_idx % input_depth;
42 const int inp_idx2 = inp_idx / input_depth;
52 const int offset_d = (offset_h * block_size + offset_w) * input_depth;
150 const int input_depth = input.dimension(3); local
156 batch_size * input_height * input_width * input_depth;
160 input_height, input_width, input_depth, output_height, output_width,
174 const int input_depth = input.dimension(1); local
185 const int input_depth_by_output_area = input_depth * output_are
33 S2D_NHWC(const int32 nthreads, const dtype* input_ptr, const int block_size, const int batch_size, const int input_height, const int input_width, const int input_depth, const int output_height, const int output_width, const int output_depth, dtype* output_ptr) argument
[all...]
H A Dconv_ops_using_gemm.cc82 // input_data = [input_batches, input_height, input_width, input_depth]
83 // filter_data = [filter_height, filter_width, input_depth, filter_count]
90 int input_depth, const T2* filter_data, int filter_height,
157 for (int in_channel = 0; in_channel < input_depth;
168 input_depth) +
169 (in_y * input_width * input_depth) +
170 (in_x * input_depth) + in_channel];
175 filter_data[(filter_y * filter_width * input_depth *
177 (filter_x * input_depth * filter_count) +
213 int input_depth, cons
88 operator ()(OpKernelContext* context, const T1* input_data, int input_batches, int input_height, int input_width, int input_depth, const T2* filter_data, int filter_height, int filter_width, int filter_count, int stride_rows, int stride_cols, Padding padding, T3* output_data, int output_height, int output_width) argument
211 operator ()(OpKernelContext* context, const T1* input_data, int input_batches, int input_height, int input_width, int input_depth, const T2* filter_data, int filter_height, int filter_width, int filter_count, int stride_rows, int stride_cols, Padding padding, T3* output_data, int output_height, int output_width) argument
221 << input_width << ", " << input_depth; local
[all...]
H A Dmkl_conv_ops.h98 int input_depth = static_cast<int>(input_depth_raw); local
120 mkldnn_sizes[MklDnnDims::Dim_C] = input_depth;
157 int input_depth = GetTensorDim(input_shape, data_format_, 'C'); local
159 OP_REQUIRES(context_, input_depth == filter_shape.dim_size(2),
161 "input and filter must have the same depth: ", input_depth,
H A Dconv_ops_fused.cc110 int input_depth; member in struct:tensorflow::__anon26385::ResizeTaskParameters
169 int64 input_width, int64 input_depth, int64 top_padding, int64 pad_offset,
184 resize_cache + (cache_index_y * cache_line_width * input_depth);
201 input_batch_start + (top_y_index * input_width * input_depth);
203 input_batch_start + (bottom_y_index * input_width * input_depth);
226 T1* cache_line_start, int64 input_depth,
233 result.cache_line_pixel = cache_line_start + (cache_index_x * input_depth);
260 int padded_height, int padded_width, int input_depth,
268 (input_depth <= 0)) {
271 << padded_width << ", " << input_depth; local
167 CalculatePerCacheLineParameters( int64 cache_height, int64 cache_y, T1* resize_cache, int64 cache_line_width, int64 input_width, int64 input_depth, int64 top_padding, int64 pad_offset, int64 resized_height, const ImageResizerState& st, const T1* input_batch_start) argument
225 CalculatePerCachePixelParameters(int64 cache_x, int64 cache_start_x, T1* cache_line_start, int64 input_depth, int64 left_padding, int64 pad_offset, int64 resized_width, const ImageResizerState& st) argument
258 operator ()(OpKernelContext* context, const Tensor& input, int input_batches, int resized_height, int resized_width, int padded_height, int padded_width, int input_depth, const T2* filter_data, int filter_height, int filter_width, int filter_count, int stride_rows, int stride_cols, Padding padding, T3* output_data, int output_height, int output_width, const ImageResizerState& st, int top_padding, int bottom_padding, int left_padding, int right_padding, int pad_offset) argument
[all...]
H A Dmkl_conv_ops.cc125 const int64 input_depth = variable
128 OP_REQUIRES(context, input_depth == filter.dim_size(2),
130 "input and filter must have the same depth: ", input_depth,
215 mkl_context.in_sizes[MklDims::C] = static_cast<size_t>(input_depth);
H A Dquantized_conv_ops.cc48 // input_data = [input_batches, input_height, input_width, input_depth]
49 // filter_data = [filter_height, filter_width, input_depth, filter_count]
56 int input_depth, int input_offset, const T2* filter_data,
134 for (int in_channel = 0; in_channel < input_depth;
145 input_depth) +
146 (in_y * input_width * input_depth) +
147 (in_x * input_depth) + in_channel];
156 filter_data[(filter_y * filter_width * input_depth *
158 (filter_x * input_depth * filter_count) +
202 int input_depth, in
54 operator ()(OpKernelContext* context, const T1* input_data, int input_batches, int input_height, int input_width, int input_depth, int input_offset, const T2* filter_data, int filter_height, int filter_width, int filter_count, int filter_offset, int stride, Padding padding, T3* output_data, int output_height, int output_width, int output_shift, int output_offset, int output_mult) argument
200 operator ()(OpKernelContext* context, const T1* input_data, int input_batches, int input_height, int input_width, int input_depth, int input_offset, const T2* filter_data, int filter_height, int filter_width, int filter_count, int filter_offset, int stride, Padding padding, T3* output_data, int output_height, int output_width, int output_shift, int output_offset, int output_mult) argument
[all...]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Dmultithreaded_conv.h117 int input_height, int input_width, int input_depth,
133 ConstEigenMatrix input(input_data, conv_width, input_depth);
134 ConstEigenMatrix filter(filter_data, input_depth, filter_count);
142 filter_width * filter_height * input_depth;
154 input_width, input_depth);
156 input_depth, filter_count);
173 const int input_depth = MatchingArraySize(input_dims, 0, filter_dims, 0); local
183 input_depth, filter_data, filter_height, filter_width,
116 operator ()(const T* input_data, T* im2col_buffer, int input_batches, int input_height, int input_width, int input_depth, const T* filter_data, int filter_height, int filter_width, int filter_count, int stride_rows, int stride_cols, int pad_width, int pad_height, TfLitePadding padding, T* output_data, int output_height, int output_width) argument
H A Ddepthwiseconv_float.h34 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
95 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
178 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
187 for (; ic <= input_depth - 16; ic += 16) {
218 for (; ic <= input_depth - 4; ic += 4) {
237 for (; ic < input_depth; ic++) {
249 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
258 for (; ic <= input_depth - 2; ic += 2) {
285 for (; ic < input_depth; ic++) {
316 // input_depth
319 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
425 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
458 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
493 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
526 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
578 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
618 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
656 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
692 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
739 Run(int num_output_pixels, int input_depth, int depth_multiplier, const float* input_ptr, int input_ptr_increment, const float* filter_ptr, float* acc_buffer_ptr) argument
764 FloatDepthwiseConvAccumRow(int stride, int input_depth, int input_width, const float* input_data, int pad_width, int depth_multiplier, int filter_width, const float* filter_data, int out_x_buffer_start, int out_x_buffer_end, int output_depth, float* acc_buffer) argument
837 FloatDepthwiseConvAccumRowGeneric( int stride, int input_depth, int input_width, const float* input_data, int pad_width, int depth_multiplier, int filter_width, const float* filter_data, int out_x_buffer_start, int out_x_buffer_end, int output_depth, float* acc_buffer) argument
923 const int input_depth = ArraySize(input_dims, 0); local
[all...]
H A Ddepthwiseconv_uint8.h34 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
81 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
149 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
219 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
296 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
362 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
476 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
536 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
631 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
701 static void Run(int num_output_pixels, int input_depth, in argument
786 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
872 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
937 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1022 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1076 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1110 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1154 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1210 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1259 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1291 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1354 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1418 Run(int num_output_pixels, int input_depth, int depth_multiplier, const uint8* input_ptr, int16 input_offset, int input_ptr_increment, const uint8* filter_ptr, int16 filter_offset, int32* acc_buffer_ptr) argument
1468 QuantizedDepthwiseConvAccumRow( int stride, int input_depth, int input_width, const uint8* input_data, int16 input_offset, int pad_width, int depth_multiplier, int filter_width, const uint8* filter_data, int16 filter_offset, int out_x_buffer_start, int out_x_buffer_end, int output_depth, int32* acc_buffer) argument
1538 QuantizedDepthwiseConvAccumRowGeneric( int stride, int input_depth, int input_width, const uint8* input_data, int16 input_offset, int pad_width, int depth_multiplier, int filter_width, const uint8* filter_data, int16 filter_offset, int out_x_buffer_start, int out_x_buffer_end, int output_depth, int32* acc_buffer) argument
1688 const int input_depth = ArraySize(input_dims, 0); local
[all...]
/external/tensorflow/tensorflow/core/kernels/neon/
H A Ddepthwiseconv_float.h38 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
99 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
182 static void Run(int num_output_pixels, int input_depth, int depth_multiplier, argument
191 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, argument
260 for (; ic <= input_depth - 2; ic += 2) {
287 for (; ic < input_depth; ic++) {
318 static void Run(int num_output_pixels, int input_depth, in argument
426 FloatDepthwiseConvAccumRow(int stride, int input_depth, int input_width, const float* input_data, int pad_width, int depth_multiplier, int filter_width, const float* filter_data, int out_x_buffer_start, int out_x_buffer_end, int output_depth, float* acc_buffer) argument
499 FloatDepthwiseConvAccumRowGeneric( int stride, int input_depth, int input_width, const float* input_data, int pad_width, int depth_multiplier, int filter_width, const float* filter_data, int out_x_buffer_start, int out_x_buffer_end, int output_depth, float* acc_buffer) argument
568 const int input_depth = ArraySize(input_dims, 0); local
[all...]
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc177 const int input_depth = weights_shape.dims(3); local
181 input_depth * kheight * kwidth});
207 const int input_depth = input_shape.dims(3); local
217 op->depth_multiplier = output_depth / input_depth;
219 QCHECK_EQ(output_depth, input_depth * op->depth_multiplier)
822 const int input_depth = input_shape.dims(input_shape.dimensions_count() - 1); local
824 CHECK_EQ(input_depth + depth, fc_input_depth);
/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dlayout_optimizer_test.cc51 int input_depth = 3; local
55 {batch_size, input_height, input_width, input_depth});
62 {filter_size, filter_size, input_depth, filter_count});
84 int input_depth = 3; local
90 {batch_size, input_height, input_width, input_depth});
95 {filter_size, filter_size, input_depth, filter_count});

Completed in 445 milliseconds

12