Home
last modified time | relevance | path

Searched refs:padding_type (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/core/framework/
Dkernel_shape_util.cc22 Padding padding_type, in GetWindowedOutputSizeVerboseV2() argument
36 switch (padding_type) { in GetWindowedOutputSizeVerboseV2()
68 int64_t stride, Padding padding_type, in GetWindowedOutputSizeVerbose() argument
74 padding_type, output_size, in GetWindowedOutputSizeVerbose()
79 int64_t stride, Padding padding_type, in GetWindowedOutputSize() argument
81 if (padding_type == Padding::EXPLICIT) { in GetWindowedOutputSize()
88 padding_type, output_size, padding_size, in GetWindowedOutputSize()
94 Padding padding_type, int64_t* output_size, in GetWindowedOutputSizeV2() argument
96 if (padding_type == Padding::EXPLICIT) { in GetWindowedOutputSizeV2()
103 stride, padding_type, output_size, in GetWindowedOutputSizeV2()
[all …]
Dkernel_shape_util.h75 int64_t stride, Padding padding_type,
104 Padding padding_type, int64_t* output_size,
115 int64_t stride, Padding padding_type,
124 Padding padding_type,
138 Padding padding_type, std::array<int64_t, 3>* output_ptr,
147 Padding padding_type,
Dcommon_shape_fns.h33 int64_t stride, Padding padding_type,
43 Padding padding_type, int64_t padding_before, int64_t padding_after,
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc46 const TfLitePadding padding_type, in ComputeSpaceToBatchParams() argument
52 if (padding_type == kTfLitePaddingSame) { in ComputeSpaceToBatchParams()
130 const TfLitePadding padding_type, TensorID* output_tensor, in BuildDilatedDwConv() argument
146 dilation_factors_h_w_, padding_type, in BuildDilatedDwConv()
244 const TfLitePadding padding_type, TensorID* output_tensor, in BuildSplittedDwConv() argument
280 if (padding_type == kTfLitePaddingSame) { in BuildSplittedDwConv()
282 } else if (padding_type == kTfLitePaddingValid) { in BuildSplittedDwConv()
332 OpBuilder* stride_node, const TfLitePadding padding_type, in BuildStandardConv() argument
342 if (padding_type == kTfLitePaddingSame) { in BuildStandardConv()
344 } else if (padding_type == kTfLitePaddingValid) { in BuildStandardConv()
[all …]
Dop_builder.h42 hexagon_nn_padding_type padding_type = NN_PAD_NA; member
94 void SetPaddingType(hexagon_nn_padding_type padding_type) { in SetPaddingType() argument
95 op_node_.padding_type = padding_type; in SetPaddingType()
297 graph_id_, op_node->node_id, op_node->op_type, op_node->padding_type, in Build()
Dconv_2d_builder.h72 const TfLitePadding padding_type,
82 const TfLitePadding padding_type,
90 const TfLitePadding padding_type, TensorID* output_tensor,
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py357 def window_padding_type_to_pad_values(padding_type, lhs_dims, rhs_dims, argument
360 if not isinstance(padding_type, (str, PaddingType)):
362 raise TypeError(msg.format(type(padding_type)))
364 if isinstance(padding_type, str):
365 if padding_type.upper() == 'VALID':
366 padding_type = PaddingType.VALID
367 elif padding_type.upper() == 'SAME':
368 padding_type = PaddingType.SAME
371 raise ValueError(msg.format(padding_type))
373 if padding_type == PaddingType.VALID:
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc245 xla::PaddingType padding_type = xla::PaddingType::PADDING_INVALID; in MakeXlaForwardConvOp() local
253 padding_type = xla::PaddingType::PADDING_VALID; in MakeXlaForwardConvOp()
256 padding_type = xla::PaddingType::PADDING_SAME; in MakeXlaForwardConvOp()
277 if (padding_type != xla::PaddingType::PADDING_INVALID) { in MakeXlaForwardConvOp()
283 /*batch_group_count=*/1, precision_config, padding_type); in MakeXlaForwardConvOp()
343 xla::PaddingType padding_type = xla::PaddingType::PADDING_INVALID; in MakeXlaBackpropInputConvOp() local
350 padding_type = xla::PaddingType::PADDING_VALID; in MakeXlaBackpropInputConvOp()
353 padding_type = xla::PaddingType::PADDING_SAME; in MakeXlaBackpropInputConvOp()
373 if (padding_type != xla::PaddingType::PADDING_INVALID) { in MakeXlaBackpropInputConvOp()
380 /*batch_group_count=*/1, precision_config, padding_type); in MakeXlaBackpropInputConvOp()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h52 DenseIndex strideCols, const PaddingType padding_type,
81 if (padding_type == PADDING_VALID) {
107 in_strideCols, padding_type,
145 DenseIndex strideCols, const PaddingType padding_type) { in CuboidMaxPooling() argument
170 if (padding_type == PADDING_VALID) { in CuboidMaxPooling()
207 strideRows, strideCols, padding_type, in CuboidMaxPooling()
377 DenseIndex strideCols, const PaddingType padding_type,
406 if (padding_type == PADDING_VALID) {
434 in_strideRows, in_strideCols, padding_type,
469 DenseIndex strideCols, const PaddingType padding_type) {
[all …]
Dpooling_ops_3d.cc101 TensorFormat data_format, Padding padding_type, in launch()
106 BrainPadding2EigenPadding(padding_type)); in launch()
116 TensorFormat data_format, Padding padding_type, in launch()
121 BrainPadding2EigenPadding(padding_type)); in launch()
801 TensorFormat data_format, Padding padding_type, in launch()
814 TensorFormat data_format, Padding padding_type, in launch()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_window_utils.cc104 PaddingType padding_type) { in GetWindowedOutputSize() argument
111 if (padding_type == PaddingType::PADDING_VALID) { in GetWindowedOutputSize()
116 } else if (padding_type == PaddingType::PADDING_SAME) { in GetWindowedOutputSize()
133 PaddingType padding_type) { in GetWindowedInputGradSize() argument
140 input_size, window_size, window_dilation, window_stride, padding_type); in GetWindowedInputGradSize()
Ddynamic_window_utils.h40 PaddingType padding_type);
46 PaddingType padding_type);
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc619 PaddingType padding_type, int pad_width, int pad_height, in TestOneDepthwiseConvWithGivenOutputShift() argument
632 op_params.padding_type = padding_type; in TestOneDepthwiseConvWithGivenOutputShift()
756 PaddingType padding_type, int pad_width, int pad_height, in TestOneDepthwiseConvBisectOutputShift() argument
771 padding_type, pad_width, pad_height, depth_multiplier, output_offset, in TestOneDepthwiseConvBisectOutputShift()
798 filter_shape, filter_offset, bias_data, bias_shape, stride, padding_type, in TestOneDepthwiseConvBisectOutputShift()
815 PaddingType padding_type, int pad_width, int pad_height, in TestOneDepthwiseConv() argument
823 filter_shape, filter_offset, bias_data, bias_shape, stride, padding_type, in TestOneDepthwiseConv()
835 PaddingType padding_type) { in TryTestDepthwiseConv() argument
884 dilation_height_factor, padding_type, in TryTestDepthwiseConv()
919 bias_data.data(), bias_shape_inference, stride, padding_type, pad_width, in TryTestDepthwiseConv()
[all …]
Dtest_util.cc29 PaddingType padding_type, RuntimeShape* output_shape, in ComputeConvSizes() argument
40 if (padding_type == PaddingType::kValid) { in ComputeConvSizes()
48 } else if (padding_type == PaddingType::kSame) { in ComputeConvSizes()
Ddepthwiseconv_float_test.cc107 const auto padding_type = in TryTestOneDepthwiseConv() local
111 dilation_height_factor, padding_type, in TryTestOneDepthwiseConv()
131 op_params.padding_type = PaddingType::kSame; in TryTestOneDepthwiseConv()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h429 Padding padding_type; variable
431 padding_type = Padding::EXPLICIT;
437 padding_type = padding_;
442 padding_type, &out_rows, &pad_top, &pad_bottom));
446 padding_type, &out_cols, &pad_left, &pad_right));
448 Padding padding_type; variable
450 padding_type = Padding::EXPLICIT;
458 padding_type = padding_;
462 stride_planes, padding_type, &out_planes,
467 padding_type, &out_rows, &pad_top, &pad_bottom));
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dpad_op_builder.h32 explicit PadOpBuilder(GraphBuilder* graph_builder, PadType padding_type) in PadOpBuilder() argument
33 : OpBuilder(graph_builder), padding_type_(padding_type) {} in PadOpBuilder()
/external/tensorflow/tensorflow/core/util/
Dpadding.cc36 Status CheckValidPadding(Padding padding_type, in CheckValidPadding() argument
39 if (padding_type == Padding::EXPLICIT) { in CheckValidPadding()
Dpadding.h50 Status CheckValidPadding(Padding padding_type,
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconv_grad_size_util.cc26 int64_t stride, Padding padding_type) { in GetWindowedOutputSize() argument
38 switch (padding_type) { in GetWindowedOutputSize()
/external/tensorflow/tensorflow/lite/kernels/
Dpooling3d.cc41 TfLitePadding padding_type; member
256 opdata->params.padding_type = kTfLitePaddingValid; in Init()
258 opdata->params.padding_type = kTfLitePaddingSame; in Init()
260 opdata->params.padding_type = kTfLitePaddingUnknown; in Init()
321 params.filter_depth, params.padding_type, &out_height, &out_width, in GenericPrepare()
Ddepthwise_conv.cc308 op_params.padding_type = PaddingType::kSame; in EvalFloat()
347 op_params.padding_type = PaddingType::kSame; in EvalQuantized()
388 op_params.padding_type = PaddingType::kSame; in EvalQuantizedPerChannel()
429 op_params.padding_type = PaddingType::kSame; in EvalQuantizedPerChannel16x8()
490 op_params.padding_type = PaddingType::kSame; in EvalHybridPerChannel()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dconv_test.cc35 const TensorData& output, Padding padding_type, in QuantizedConvolutionOpModel() argument
86 builder_, padding_type, stride_length, stride_length, depth_mul, in QuantizedConvolutionOpModel()
91 CreateConv2DOptions(builder_, padding_type, stride_length, in QuantizedConvolutionOpModel()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dtypes.h44 static ::tflite::Padding Serialize(PaddingType padding_type);
Dtypes.cc203 ::tflite::Padding Padding::Serialize(PaddingType padding_type) { in Serialize() argument
204 switch (padding_type) { in Serialize()

123