Home
last modified time | relevance | path

Searched refs:pad_params (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dimage_preprocessing_stage.cc189 tflite::PadParams pad_params; in Pad() local
190 pad_params.left_padding_count = 4; in Pad()
191 std::uninitialized_fill_n(pad_params.left_padding, 4, 0); in Pad()
192 pad_params.left_padding[1] = in Pad()
194 pad_params.left_padding[2] = in Pad()
196 pad_params.right_padding_count = 4; in Pad()
197 std::uninitialized_fill_n(pad_params.right_padding, 4, 0); in Pad()
198 pad_params.right_padding[1] = in Pad()
199 output_height - pad_params.left_padding[1] - image_data->height; in Pad()
200 pad_params.right_padding[2] = in Pad()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_operator.cc40 StatusOr<mlir::StringAttr> GetPaddingAttr(TfLitePadding pad_params, in GetPaddingAttr() argument
44 if (pad_params == TfLitePadding::kTfLitePaddingSame) { in GetPaddingAttr()
46 } else if (pad_params == TfLitePadding::kTfLitePaddingValid) { in GetPaddingAttr()
50 absl::StrCat("Invalid padding type", std::to_string(pad_params))); in GetPaddingAttr()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc54 TfLitePadParams pad_params; in TEST() local
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc345 TfLitePadParams* pad_params = in BuildPadLoopBodySubgraph() local
350 {}, nullptr, 0, pad_params, pad_reg, in BuildPadLoopBodySubgraph()