Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dpadded_batch_dataset_op.cc52 std::vector<Tensor> padding_values, const DatasetBase* input, in Dataset()
142 std::vector<Node*> padding_values; in AsGraphDefInternal() local
426 std::vector<Tensor> padding_values; in MakeDataset() local
/external/tensorflow/tensorflow/lite/kernels/
Dpadding.h66 TfLitePaddingValues padding_values; in ComputePaddingHeightWidth() local
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dpadded_batch_test.py135 def testPaddedBatchDatasetNonDefaultPadding(self, padding_values): argument
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc28 TfLitePaddingValues padding_values; in TEST() local
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py132 padding_values=None, argument
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py1379 padding_values=None, argument
2297 padding_values=None, argument
3786 def _padding_values_or_default(padding_values, input_dataset): argument
3811 def __init__(self, input_dataset, batch_size, padded_shapes, padding_values, argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc444 auto padding_values = in Compile() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h811 PaddingValues padding_values; member
838 PaddingValues padding_values; member
976 PaddingValues padding_values; member
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1971 std::vector<std::pair<int64, int64>> padding_values = in ReduceWindow() local