Home
last modified time | relevance | path

Searched defs:padding_values (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dpadded_batch_dataset_op.cc54 std::vector<Tensor> padding_values, const DatasetBase* input, in Dataset()
154 std::vector<Node*> padding_values; in AsGraphDefInternal() local
459 std::vector<Tensor> padding_values; in MakeDataset() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dpadding.h71 TfLitePaddingValues padding_values; in ComputePaddingHeightWidth() local
97 Padding3DValues padding_values; in ComputePadding3DValues() local
H A Dpooling3d.cc42 Padding3DValues padding_values; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtypes.h699 PaddingValues padding_values; member
721 Padding3DValues padding_values; member
741 PaddingValues padding_values; member
888 PaddingValues padding_values; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dbuiltin_op_data_test.cc28 TfLitePaddingValues padding_values; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dgrouping.py117 padding_values=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py1904 padding_values=None, argument
2975 padding_values=None, argument
3991 padding_values=None, argument
5284 def _padding_values_or_default(padding_values, input_dataset): argument
5316 padding_values, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dpadded_batch_test.py154 def testPaddedBatchDatasetNonDefaultPadding(self, padding_values): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
H A Dtf_to_xla_attribute_utils.cc174 TfLitePaddingValues padding_values = tflite::ComputePaddingHeightWidth( in CalculatePaddingAndPadIfNeeded() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpooling_ops.cc500 auto padding_values = in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc245 SmallVector<int32_t> padding_values(padding_attr_array.size()); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference_test.cc1146 std::vector<std::pair<int64_t, int64_t>> padding_values = in TEST_F() local
1208 std::vector<std::pair<int64_t, int64_t>> padding_values = in TEST_F() local
/aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
H A Donedevice_train_palm_detection_filter_inference_beta.cc376 const TPaddings* __restrict padding_values, in SpaceToBatchNDRank4()
H A Donedevice_train_palm_detection_filter_inference.cc377 const TPaddings* __restrict padding_values, in SpaceToBatchNDRank4()
H A Donedevice_train_palm_detection_filter_inference_v2.cc376 const TPaddings* __restrict padding_values, in SpaceToBatchNDRank4()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc945 auto padding_values = padding.getValues<int64_t>(); in MatchResizeOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc2666 std::vector<std::pair<int64_t, int64_t>> padding_values = in ReduceWindow() local
3265 std::vector<std::pair<int64_t, int64_t>> padding_values = in SelectAndScatter() local