| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | cell_wrapper.py | 170 def construct(self, input_ids, input_mask, token_type_id, label_ids): argument
184 def construct(self, input_ids, input_mask, token_type_id, label_ids): argument
255 def construct(self, input_ids, input_mask, token_type_id, label_ids): argument
288 def construct(self, input_ids, input_mask, token_type_id, label_ids): argument
|
| D | model.py | 153 def construct(self, input_ids): argument
749 def construct(self, input_ids, token_type_ids, input_mask): argument
828 def construct(self, input_ids, input_mask, token_type_id): argument
861 def construct(self, input_ids, input_mask, token_type_id): argument
887 def construct(self, input_ids, input_mask, token_type_id): argument
|
| D | dataset.py | 28 def __init__(self, input_ids, input_mask, segment_ids, label_id, seq_length=None): argument
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | bert_for_pre_training.py | 165 def construct(self, input_ids, input_mask, token_type_id, argument
243 input_ids, argument
293 input_ids, argument
382 input_ids, argument
|
| D | bert_model.py | 138 def construct(self, input_ids): argument
915 def construct(self, input_ids, token_type_ids, input_mask): argument
|
| /third_party/mindspore/tests/ut/data/dataset/testTFBert5Rows2/ |
| D | datasetSchema.json | 5 "input_ids": { object
|
| /third_party/mindspore/tests/ut/data/dataset/testTFBert5Rows/ |
| D | datasetSchema.json | 30 "input_ids": { object
|
| /third_party/mindspore/tests/mindspore_test_framework/apps/ |
| D | test_bert_parts.py | 86 input_ids, input_mask, token_type_id, \ variable
|
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | tinybert_train_export.py | 158 input_ids = M.Tensor(input_ids_np, M.int32) variable
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 537 def construct(self, input_ids): argument
|