| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_value.py | 35 def __init__(self, values, row_splits): argument
54 row_splits = property( variable in RaggedTensorValue
|
| D | ragged_factory_ops.py | 76 def ragged_factory(values, row_splits): argument
135 def _ragged_factory(values, row_splits): argument
|
| D | row_partition.py | 115 row_splits, argument
312 def from_row_splits(cls, argument
733 def row_splits(self): member in RowPartition
|
| D | ragged_tensor.py | 413 def from_row_splits(cls, values, row_splits, name=None, validate=True): argument
999 def row_splits(self): member in RaggedTensor
|
| D | ragged_tensor_test.py | 2160 def test_fn(flat_value, row_splits): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_array_ops.cc | 98 ShapeHandle row_splits = c->input(i + ragged_splits_start); in __anon8359b0a00102() local
118 ShapeHandle row_splits = c->Vector(batch_size + 1); in __anon8359b0a00102() 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 | 250 Tensor row_splits = context->input(1); in Compute() local
|
| D | ragged_cross_op.cc | 79 RaggedFeatureReader(const Tensor& values, const Tensor& row_splits) in RaggedFeatureReader()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils.cc | 156 int row_splits = func.getFunctionType().getInputs().size() - kRowSplits; in VerifyNgrams() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2.py | 921 def _add_data_for_ragged_tensor(self, tensor, weight, row_splits, values, argument
|
| D | tpu_embedding.py | 227 row_splits=None, argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1331 Tensor* row_splits = &result->ragged_splits.back(); in FastParseExample() local
|