Searched refs:mask_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention_test.py | 92 mask_tensor = keras.Input(shape=(4, 2))
93 output = test_layer(query=query, value=value, attention_mask=mask_tensor)
96 model = keras.Model([query, value, mask_tensor], output)
117 output = test_layer(query, value=value, key=key, attention_mask=mask_tensor)
118 model = keras.Model([query, value, key, mask_tensor], output)
153 mask_tensor = keras.Input(shape=(4, 2))
154 output = test_layer(query=query, value=value, attention_mask=mask_tensor)
157 model = keras.Model([query, value, mask_tensor], output)
178 query=query, value=value, attention_mask=mask_tensor,
180 model = keras.Model([query, value, mask_tensor], [output, scores])
[all …]
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| D | bert_nl_classifier.cc | 70 auto* mask_tensor = in Preprocess() local
112 PopulateTensor(input_mask, mask_tensor); in Preprocess()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/qa/ |
| D | bert_question_answerer.cc | 171 TfLiteTensor* mask_tensor = in Preprocess() local
279 PopulateTensor(input_mask, mask_tensor); in Preprocess()
|