Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dpad.cc50 const int32* paddings_data = GetTensorData<int32>(paddings); in PadContext() local
54 (paddings_data[0] == 0 && paddings_data[1] == 0) && in PadContext()
55 (paddings_data[6] == 0 && paddings_data[7] == 0)) { in PadContext()
80 const int32* paddings_data = GetTensorData<int32>(op_context->paddings); in ResizeOutputTensor() local
83 int before_padding = *paddings_data++; in ResizeOutputTensor()
84 int after_padding = *paddings_data++; in ResizeOutputTensor()
138 const int32* paddings_data = GetTensorData<int32>(op_context.paddings); in Eval() local
145 before_padding.push_back(paddings_data[idx * 2]); in Eval()
146 after_padding.push_back(paddings_data[idx * 2 + 1]); in Eval()
Dspace_to_batch_nd.cc60 const int32* paddings_data = GetTensorData<int32>(op_context->paddings); in ResizeOutputTensor() local
74 int final_dim_size = (input_size->data[dim + 1] + paddings_data[dim * 2] + in ResizeOutputTensor()
75 paddings_data[dim * 2 + 1]); in ResizeOutputTensor()
/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/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1308 const auto& paddings_data = in ProcessSpaceToBatchNDOperator() local
1310 int height_with_paddings = input_height + paddings_data[0] + paddings_data[1]; in ProcessSpaceToBatchNDOperator()
1311 int width_with_paddings = input_width + paddings_data[2] + paddings_data[3]; in ProcessSpaceToBatchNDOperator()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc61 auto paddings_data = paddings_t->matrix<T>(); in PadKnown() local
63 const T pad0 = paddings_data(i, 0); in PadKnown()
64 const T pad1 = paddings_data(i, 1); in PadKnown()
1809 auto paddings_data = paddings_t->matrix<T>(); in MirrorPadKnown() local
1812 const int64 pad0 = static_cast<int64>(paddings_data(i, 0)); in MirrorPadKnown()
1813 const int64 pad1 = static_cast<int64>(paddings_data(i, 1)); in MirrorPadKnown()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h1922 const int32* paddings_data, in SpaceToBatchND() argument
1931 DimsToShape(paddings_dims), paddings_data, in SpaceToBatchND()
1939 const int32* paddings_data, in SpaceToBatchND() argument
1947 DimsToShape(paddings_dims), paddings_data, in SpaceToBatchND()
Dreference_ops.h3079 const RuntimeShape& unextended_input3_shape, const int32* paddings_data, in SpaceToBatchND() argument
3100 const int padding_top = paddings_data[0]; in SpaceToBatchND()
3101 const int padding_left = paddings_data[2]; in SpaceToBatchND()