Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad.cc51 const TfLiteTensor* padding_matrix = nullptr; member
73 inline void GetPadding(const TfLiteTensor* padding_matrix, int dimension, in GetPadding() argument
75 switch (padding_matrix->type) { in GetPadding()
77 GetPadding(padding_matrix->data.i32, dimension, left_pad, right_pad); in GetPadding()
80 GetPadding(padding_matrix->data.i64, dimension, left_pad, right_pad); in GetPadding()
112 const TfLiteTensor* padding_matrix = eval_data->padding_matrix; in Eval() local
116 GetPadding(padding_matrix, current_dim, &left_pad, &right_pad); in Eval()
140 const TfLiteTensor* input, const TfLiteTensor* padding_matrix) { in GetPaddedOutputShape() argument
147 GetPadding(padding_matrix, i, &left_pad, &right_pad); in GetPaddedOutputShape()
157 const TfLiteTensor* padding_matrix = GetInput(context, node, 1); in Eval() local
[all …]
Dmirror_pad_test.cc30 const TensorData& padding_matrix, in BaseMirrorPadOpModel() argument
34 padding_matrix_id_ = AddInput(padding_matrix); in BaseMirrorPadOpModel()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc2007 const auto& padding_matrix = model->GetArray(op->inputs[1]); in ProcessMirrorPadOperator() local
2025 if (padding_matrix.data_type == ArrayDataType::kInt32) { in ProcessMirrorPadOperator()
2026 const auto& data = padding_matrix.GetBuffer<ArrayDataType::kInt32>().data; in ProcessMirrorPadOperator()
2030 } else if (padding_matrix.data_type == ArrayDataType::kInt64) { in ProcessMirrorPadOperator()
2031 const auto& data = padding_matrix.GetBuffer<ArrayDataType::kInt64>().data; in ProcessMirrorPadOperator()
2036 CHECK(padding_matrix.data_type == ArrayDataType::kInt64 || in ProcessMirrorPadOperator()
2037 padding_matrix.data_type == ArrayDataType::kInt32); in ProcessMirrorPadOperator()
2039 CHECK_EQ(padding_matrix.shape().dimensions_count(), 2); in ProcessMirrorPadOperator()
2041 padding_matrix.shape().dims(0)); in ProcessMirrorPadOperator()
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py4153 padding_matrix = tf.placeholder(
4157 input_tensors = [input_tensor, padding_matrix]
4159 padding_matrix = tf.constant(np.array(parameters["padding_matrix"]))
4162 input_tensor, paddings=padding_matrix, mode=parameters["mode"])