Searched defs:row_splits (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_value.py | 36 def __init__(self, values, row_splits): argument
55 row_splits = property( variable in RaggedTensorValue
|
| D | ragged_factory_ops.py | 80 def ragged_factory(values, row_splits): argument
139 def _ragged_factory(values, row_splits): argument
|
| D | row_partition.py | 113 row_splits, argument
302 def from_row_splits(cls, row_splits, validate=True, preferred_dtype=None): argument
653 def row_splits(self): member in RowPartition
|
| D | ragged_tensor.py | 407 def from_row_splits(cls, values, row_splits, name=None, validate=True): argument
930 def row_splits(self): member in RaggedTensor
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_array_ops.cc | 91 ShapeHandle row_splits = c->input(i + ragged_splits_start); in __anon9deb1dc20102() local
104 ShapeHandle row_splits = c->Vector(batch_size + 1); in __anon9deb1dc20102() local
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | whitespace_tokenizer.cc | 126 TfLiteTensor* row_splits = nested_row_splits[i]; in WriteRaggedOutput() local
133 TfLiteTensor* row_splits = nested_row_splits.back(); in WriteRaggedOutput() local
173 TfLiteTensor* row_splits = in Prepare() local
|
| D | ngrams_test.cc | 83 std::vector<int64_t> row_splits; in NgramsModel() local
120 std::vector<int64_t> row_splits = in ExtractRowLengthsTensorVector() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_variant_op.cc | 189 Tensor row_splits = context->input(1); in Compute() local
|
| D | ragged_cross_op.cc | 78 RaggedFeatureReader(const Tensor& values, const Tensor& row_splits) in RaggedFeatureReader()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils.cc | 157 int row_splits = func.getType().getInputs().size() - kRowSplits; in VerifyNgrams() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1315 Tensor* row_splits = &result->ragged_splits.back(); in FastParseExample() local
|