| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| H A D | tflite_text_api.py | 58 string_separator=' ', argument
101 'attr { key: "string_separator" value { s: "%s" } }' % string_separator,
117 string_separator, name)
123 return tf_text.ngrams(data, width, axis, reduction_type, string_separator,
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| H A D | ngrams_test.cc | 43 NgramsModel(int width, const std::string& string_separator, in NgramsModel() argument
49 BuildCustomOp(width, string_separator); in NgramsModel()
59 NgramsModel(int width, const std::string& string_separator, in NgramsModel() argument
78 BuildCustomOp(width, string_separator); in NgramsModel()
134 void BuildCustomOp(int width, const std::string& string_separator) { in BuildCustomOp() argument
138 fbb.String("string_separator", string_separator); in BuildCustomOp()
|
| H A D | ngrams.cc | 60 std::string string_separator; member
66 string_separator(m["string_separator"].ToString()) {} in NgramsAttributes()
171 separator.str = attributes.string_separator.c_str(); in Eval()
172 separator.len = attributes.string_separator.length(); in Eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | tftext_utils.cc | 219 auto string_separator = in CreateNgramsCustomOption() local
221 if (!string_separator) { in CreateNgramsCustomOption()
227 std::string string_separator_str(string_separator.getValue().data(), in CreateNgramsCustomOption()
228 string_separator.getValue().size()); in CreateNgramsCustomOption()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A D | metrics_nonportable_test.py | 331 def mock_ngrams(data, width, axis=-1, string_separator=' ', name=None): argument
338 'attr { key: "string_separator" value { s: "%s" } }' % string_separator,
360 windowed_data, axis=axis, separator=string_separator)
418 return mock_ngrams(input_tensor, width=2, axis=-1, string_separator=' ')
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/metrics/ |
| H A D | metrics_nonportable_test.py | 331 def mock_ngrams(data, width, axis=-1, string_separator=' ', name=None): argument
338 'attr { key: "string_separator" value { s: "%s" } }' % string_separator,
360 windowed_data, axis=axis, separator=string_separator)
420 return mock_ngrams(input_tensor, width=2, axis=-1, string_separator=' ')
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| H A D | fuse-tftext.mlir | 3197 …@"tftext:Ngrams", {axis = -1 : i64, reduction_type = "STRING_JOIN", string_separator = " ", width …
3211 …@"tftext:Ngrams", {axis = -1 : i64, reduction_type = "STRING_JOIN", string_separator = " ", width …
3216 …@"tftext:Ngrams", {axis = -1 : i64, reduction_type = "STRING_JOIN", string_separator = "", width =…
3436 …@"tftext:Ngrams", {axis = -1 : i64, reduction_type = "STRING_JOIN", string_separator = "", width =…
|