Searched defs:padding_values (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | padded_batch_dataset_op.cc | 52 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/ |
D | padding.h | 66 TfLitePaddingValues padding_values; in ComputePaddingHeightWidth() local
|
/external/tensorflow/tensorflow/python/data/kernel_tests/ |
D | padded_batch_test.py | 135 def testPaddedBatchDatasetNonDefaultPadding(self, padding_values): argument
|
/external/tensorflow/tensorflow/lite/c/ |
D | builtin_op_data_test.cc | 28 TfLitePaddingValues padding_values; in TEST() local
|
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | grouping.py | 132 padding_values=None, argument
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | dataset_ops.py | 1379 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/ |
D | pooling_ops.cc | 444 auto padding_values = in Compile() local
|
/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | types.h | 811 PaddingValues padding_values; member 838 PaddingValues padding_values; member 976 PaddingValues padding_values; member
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 1971 std::vector<std::pair<int64, int64>> padding_values = in ReduceWindow() local
|