| /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 | 98 l2pool_op->kwidth = avpool_op->kwidth; in Run()
|
| D | propagate_fixed_sizes.cc | 34 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes() argument
46 CHECK_GE(kwidth, 1); in ComputeConvSizes()
53 int dilated_kwidth = dilation_width_factor * (kwidth - 1) + 1; in ComputeConvSizes()
146 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
147 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width, in ProcessConvOperator()
161 input_depth * kheight * kwidth}); in ProcessConvOperator()
210 const int kwidth = weights_shape.dims(2); in ProcessTransposeConvOperator() local
217 op->padding.fixed->width = (kwidth - 1) / 2; in ProcessTransposeConvOperator()
249 specified_output_shape[2], input_depth * kheight * kwidth}); in ProcessTransposeConvOperator()
287 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 | 51 op.stride_height, op.kwidth, in WriteOptions()
60 op->kwidth = options.filter_width(); in ReadOptions()
648 op.stride_height, op.kwidth, in WriteOptions()
657 op->kwidth = options.filter_width(); in ReadOptions()
706 op.stride_height, op.kwidth, in WriteOptions()
715 op->kwidth = options.filter_width(); in ReadOptions()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 449 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn() argument
454 DimsToShape(input_dims), w, h, b, kheight, kwidth, stride_width, in ExtractPatchIntoBufferColumn()
484 int kwidth, uint8 zero_byte, T* output_data, in Im2col() argument
496 Im2col(op_params, kheight, kwidth, zero_byte, DimsToShape(input_dims), in Im2col()
503 int pad_width, int pad_height, int kheight, int kwidth, in Im2col() argument
506 kwidth, zero_byte, output_data, output_dims); in Im2col()
711 int pad_width, int pad_height, int kheight, int kwidth, in Im2col() argument
714 kwidth, zero_byte, output_data, output_dims); in Im2col()
1204 int pad_height, int kwidth, int kheight, in AveragePool() argument
1212 params.filter_width = kwidth; in AveragePool()
[all …]
|
| D | optimized_ops.h | 1974 int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn() argument
1985 const int kwidth_times_indepth = kwidth * in_depth; in ExtractPatchIntoBufferColumn()
1991 const int iw_ungated_end = (iw_ungated_start + kwidth); in ExtractPatchIntoBufferColumn()
2000 std::min(kwidth - w_offset, in_width - iw_start) * in_depth; in ExtractPatchIntoBufferColumn()
2003 output_row_offset + (h_offset * kwidth + w_offset) * in_depth; in ExtractPatchIntoBufferColumn()
2012 ((kwidth - (left_padding + right_padding)) * in_depth)); in ExtractPatchIntoBufferColumn()
2017 const int top_row_elements = (top_padding * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
2053 const int bottom_row_elements = (bottom_padding * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
2056 ((top_padding + (ih_end - ih_start)) * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
2146 void Im2col(const ConvParams& params, int kheight, int kwidth, uint8 zero_byte, in Im2col() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 867 int kwidth = 0;
898 int kwidth = 0;
913 int kwidth = 0;
|
| D | tooling_util.cc | 1939 maxpool.kheight * maxpool.kwidth; in EstimateArithmeticOpsCount()
1949 avgpool.kheight * avgpool.kwidth; in EstimateArithmeticOpsCount()
1960 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 | 1038 maxpool->kwidth = ksize.i(2); in ConvertMaxPoolOperator()
1077 avgpool->kwidth = ksize.i(2); in ConvertAvgPoolOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 1400 int pad_height, int kwidth, int kheight, in AveragePool() argument
1408 params.filter_width = kwidth; in AveragePool()
1475 int pad_height, int kwidth, int kheight, float* output_data, in AveragePool() argument
1481 pad_height, kwidth, kheight, output_activation_min, in AveragePool()
1549 int pad_height, int kwidth, int kheight, in MaxPool() argument
1556 params.filter_width = kwidth; in MaxPool()
1569 int kwidth, int kheight, float* output_data, in MaxPool() argument
1574 pad_height, kwidth, kheight, output_activation_min, in MaxPool()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1837 int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn() argument
1848 const int kwidth_times_indepth = kwidth * in_depth; in ExtractPatchIntoBufferColumn()
1854 const int iw_ungated_end = (iw_ungated_start + kwidth); in ExtractPatchIntoBufferColumn()
1863 std::min(kwidth - w_offset, in_width - iw_start) * in_depth; in ExtractPatchIntoBufferColumn()
1866 output_row_offset + (h_offset * kwidth + w_offset) * in_depth; in ExtractPatchIntoBufferColumn()
1875 ((kwidth - (left_padding + right_padding)) * in_depth)); in ExtractPatchIntoBufferColumn()
1880 const int top_row_elements = (top_padding * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
1916 const int bottom_row_elements = (bottom_padding * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
1919 ((top_padding + (ih_end - ih_start)) * kwidth * in_depth); in ExtractPatchIntoBufferColumn()
1929 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn() argument
[all …]
|