/external/tensorflow/tensorflow/contrib/lite/kernels/ |
H A D | depthwise_conv_test.cc | 57 int input_depth = GetShape(input_)[3]; local 59 int depth_mul = output_depth / input_depth;
|
H A D | conv.cc | 211 // [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 D | cxx11_tensor_image_patch.cpp | 185 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 D | depthwiseconv_float.h | 37 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 D | depthwiseconv_uint8.h | 43 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 D | reference_ops.h | 168 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 D | quantized_batch_norm_op_test.cc | 66 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 D | conv_ops_test.cc | 162 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 D | depthtospace_op.cc | 88 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 D | depthtospace_op_gpu.cu.cc | 38 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 D | eigen_backward_spatial_convolutions_test.cc | 32 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 D | eigen_spatial_convolutions_test.cc | 28 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 D | spacetodepth_op.cc | 88 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 D | spacetodepth_op_gpu.cu.cc | 36 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 D | conv_ops_using_gemm.cc | 82 // 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 D | mkl_conv_ops.h | 98 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 D | conv_ops_fused.cc | 110 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 D | mkl_conv_ops.cc | 125 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 D | quantized_conv_ops.cc | 48 // 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 D | multithreaded_conv.h | 117 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 D | depthwiseconv_float.h | 34 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 D | depthwiseconv_uint8.h | 34 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 D | depthwiseconv_float.h | 38 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 D | propagate_fixed_sizes.cc | 177 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 D | layout_optimizer_test.cc | 51 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});
|