Searched refs:paddings_data (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | pad.cc | 50 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()
|
D | space_to_batch_nd.cc | 60 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/ |
D | flatten_atrous_test.cc | 55 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/ |
D | propagate_fixed_sizes.cc | 1308 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/ |
D | array_ops.cc | 61 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/ |
D | legacy_reference_ops.h | 1922 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()
|
D | reference_ops.h | 3079 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()
|