Home
last modified time | relevance | path

Searched refs:paddings_data (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dpad.cc55 const int32* paddings_data = GetTensorData<int32>(paddings); in PadContext() local
59 (paddings_data[0] == 0 && paddings_data[1] == 0) && in PadContext()
60 (paddings_data[6] == 0 && paddings_data[7] == 0)) { in PadContext()
83 const int32* paddings_data = GetTensorData<int32>(op_context->paddings); in ResizeOutputTensor() local
86 int before_padding = *paddings_data++; in ResizeOutputTensor()
87 int after_padding = *paddings_data++; in ResizeOutputTensor()
96 paddings_data = GetTensorData<int32>(op_context->paddings); in ResizeOutputTensor()
99 int before_padding = *paddings_data++; in ResizeOutputTensor()
100 int after_padding = *paddings_data++; in ResizeOutputTensor()
189 const int32* paddings_data = GetTensorData<int32>(op_context.paddings); in Eval() local
[all …]
Dspace_to_batch_nd.cc61 const int32* paddings_data = GetTensorData<int32>(op_context->paddings); in ResizeOutputTensor() local
80 int final_dim_size = (input_size->data[dim + 1] + paddings_data[dim * 2] + in ResizeOutputTensor()
81 paddings_data[dim * 2 + 1]); in ResizeOutputTensor()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dpad.cc81 const int32_t* paddings_data = GetTensorData<int32_t>(paddings); in Prepare() local
85 input->dims->data[i] + paddings_data[i * 2] + paddings_data[i * 2 + 1]; in Prepare()
92 if (paddings_total == 8 && (paddings_data[0] == 0 && paddings_data[1] == 0) && in Prepare()
93 (paddings_data[6] == 0 && paddings_data[7] == 0)) { in Prepare()
102 data->params.left_padding[idx] = paddings_data[idx * 2]; in Prepare()
103 data->params.right_padding[idx] = paddings_data[idx * 2 + 1]; in Prepare()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dspace_to_batch_nd.h46 const int32_t* paddings_data, in SpaceToBatchND() argument
73 const int padding_top = paddings_data[0]; in SpaceToBatchND()
75 unextended_input1_shape.DimensionsCount() == 4 ? paddings_data[2] : 0; in SpaceToBatchND()
Dlegacy_reference_ops.h2014 const int32* paddings_data, in SpaceToBatchND() argument
2023 DimsToShape(paddings_dims), paddings_data, in SpaceToBatchND()
2031 const int32* paddings_data, in SpaceToBatchND() argument
2039 DimsToShape(paddings_dims), paddings_data, in SpaceToBatchND()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dflatten_atrous_test.cc55 Tensor paddings_data(DT_INT32, TensorShape({2, 2})); in TestFlattenAtrousConv() local
56 test::FillValues<int>(&paddings_data, {1, 2, 1, 2}); in TestFlattenAtrousConv()
58 Input::Initializer(paddings_data)); in TestFlattenAtrousConv()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc68 auto paddings_data = paddings_t->matrix<T>(); in PadKnown() local
70 const T pad0 = paddings_data(i, 0); in PadKnown()
71 const T pad1 = paddings_data(i, 1); in PadKnown()
1964 auto paddings_data = paddings_t->matrix<T>(); in MirrorPadKnown() local
1967 const int64 pad0 = static_cast<int64>(paddings_data(i, 0)); in MirrorPadKnown()
1968 const int64 pad1 = static_cast<int64>(paddings_data(i, 1)); in MirrorPadKnown()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc2399 const int32_t* paddings_data = in VisitPadNode() local
2402 const int32_t pre_padding = paddings_data[i * 2 + 0]; in VisitPadNode()
2411 const int32_t post_padding = paddings_data[i * 2 + 1]; in VisitPadNode()
2425 pre_paddings[i] = static_cast<size_t>(paddings_data[i * 2 + 0]); in VisitPadNode()
2426 post_paddings[i] = static_cast<size_t>(paddings_data[i * 2 + 1]); in VisitPadNode()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1318 const auto& paddings_data = in ProcessSpaceToBatchNDOperator() local
1325 int final_dim_size = (input_shape.dims(dim + 1) + paddings_data[dim * 2] + in ProcessSpaceToBatchNDOperator()
1326 paddings_data[dim * 2 + 1]); in ProcessSpaceToBatchNDOperator()