| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | im2col_utils.h | 26 int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn() argument
37 const int kwidth_times_indepth = kwidth * in_depth; in ExtractPatchIntoBufferColumn()
43 const int iw_ungated_end = (iw_ungated_start + kwidth); in ExtractPatchIntoBufferColumn()
52 std::min(kwidth - w_offset, in_width - iw_start) * in_depth; in ExtractPatchIntoBufferColumn()
55 output_row_offset + (h_offset * kwidth + w_offset) * in_depth; in ExtractPatchIntoBufferColumn()
64 ((kwidth - (left_padding + right_padding)) * in_depth)); in ExtractPatchIntoBufferColumn()
69 const int top_row_elements = (top_padding * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
105 const int bottom_row_elements = (bottom_padding * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
108 ((top_padding + (ih_end - ih_start)) * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
198 void Im2col(const ConvParams& params, int kheight, int kwidth, uint8 zero_byte, in Im2col() argument
[all …]
|
| D | legacy_optimized_ops.h | 2361 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn() argument
2366 DimsToShape(input_dims), w, h, b, kheight, kwidth, stride_width, in ExtractPatchIntoBufferColumn()
2396 int kwidth, uint8 zero_byte, T* output_data, in Im2col() argument
2408 Im2col(op_params, kheight, kwidth, zero_byte, DimsToShape(input_dims), in Im2col()
2415 int pad_width, int pad_height, int kheight, int kwidth, in Im2col() argument
2418 kwidth, zero_byte, output_data, output_dims); in Im2col()
2835 int pad_width, int pad_height, int kheight, int kwidth, in Im2col() argument
2838 kwidth, zero_byte, output_data, output_dims); in Im2col()
3754 int pad_height, int kwidth, int kheight, in AveragePool() argument
3762 params.filter_width = kwidth; in AveragePool()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | create_im2col_arrays.cc | 42 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
43 if (kwidth == 1 && kheight == 1 && op->stride_width == 1 && in ProcessConvOperator()
|
| D | identify_l2_pool.cc | 84 l2pool_op->kwidth = avpool_op->kwidth; in Run()
|
| D | propagate_fixed_sizes.cc | 35 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes() argument
47 CHECK_GE(kwidth, 1); in ComputeConvSizes()
54 int dilated_kwidth = dilation_width_factor * (kwidth - 1) + 1; in ComputeConvSizes()
147 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
148 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width, in ProcessConvOperator()
162 input_depth * kheight * kwidth}); in ProcessConvOperator()
211 const int kwidth = weights_shape.dims(2); in ProcessTransposeConvOperator() local
218 op->padding.fixed->width = (kwidth - 1) / 2; in ProcessTransposeConvOperator()
250 specified_output_shape[2], input_depth * kheight * kwidth}); in ProcessTransposeConvOperator()
288 const int kwidth = weights_shape.dims(2); in ProcessDepthwiseConvOperator() local
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 264 op.kwidth = 480; in TEST_F()
271 EXPECT_EQ(op.kwidth, output_toco_op->kwidth); in TEST_F()
296 op.kwidth = 480; in TEST_F()
303 EXPECT_EQ(op.kwidth, output_toco_op->kwidth); in TEST_F()
353 op.kwidth = 480; in TEST_F()
362 EXPECT_EQ(op.kwidth, output_toco_op->kwidth); in TEST_F()
|
| D | operator.cc | 98 op.stride_height, op.kwidth, in WriteOptions()
107 op->kwidth = options.filter_width(); in ReadOptions()
523 op.stride_height, op.kwidth, in WriteOptions()
532 op->kwidth = options.filter_width(); in ReadOptions()
573 op.stride_height, op.kwidth, in WriteOptions()
582 op->kwidth = options.filter_width(); in ReadOptions()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 911 int kwidth = 0;
942 int kwidth = 0;
957 int kwidth = 0;
|
| D | tooling_util.cc | 1975 maxpool.kheight * maxpool.kwidth; in EstimateArithmeticOpsCount()
1985 avgpool.kheight * avgpool.kwidth; in EstimateArithmeticOpsCount()
1996 const int64 cost_per_val = 2 * maxpool->kheight * maxpool->kwidth + 32; in EstimateArithmeticOpsCount()
|
| D | export_tensorflow.cc | 991 ksize.mutable_list()->add_i(src_op.kwidth); in ConvertMaxPoolOperator()
1019 ksize.mutable_list()->add_i(src_op.kwidth); in ConvertAveragePoolOperator()
1096 ksize.mutable_list()->add_i(src_op.kwidth); in ConvertL2PoolOperator()
|
| D | import_tensorflow.cc | 1253 maxpool->kwidth = ksize.i(2); in ConvertMaxPoolOperator()
1292 avgpool->kwidth = ksize.i(2); in ConvertAvgPoolOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 1521 int pad_height, int kwidth, int kheight, in AveragePool() argument
1529 params.filter_width = kwidth; in AveragePool()
1596 int pad_height, int kwidth, int kheight, float* output_data, in AveragePool() argument
1602 pad_height, kwidth, kheight, output_activation_min, in AveragePool()
1670 int pad_height, int kwidth, int kheight, in MaxPool() argument
1677 params.filter_width = kwidth; in MaxPool()
1690 int kwidth, int kheight, float* output_data, in MaxPool() argument
1695 pad_height, kwidth, kheight, output_activation_min, in MaxPool()
|