| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/bert/bert_performance/src/ |
| D | bert_for_pre_training.py | 166 def construct(self, input_ids, input_mask, token_type_id, argument
246 input_ids, argument
282 input_ids, argument
361 input_ids, argument
435 input_ids, argument
583 input_ids, argument
729 input_ids, argument
834 input_ids, argument
891 input_ids, argument
|
| D | bert_model.py | 126 def construct(self, input_ids): argument
830 def construct(self, input_ids, token_type_ids, input_mask): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_autoparallel_rec_shared_param_strmodif.py | 31 def __init__(self, input_ids, length=3): argument
68 def construct(self, input_ids): argument
|
| D | test_asd.py | 79 def construct(self, input_ids): argument
159 def construct(self, input_ids): argument
|
| D | test_dynamic_pangu_alpha_without_skip_redistribution.py | 292 def construct(self, input_ids): argument
825 …def construct(self, input_ids, input_mask, input_position=None, attention_mask=None, layer_past=No… argument
917 def construct(self, input_ids, input_mask, input_position=None, attention_mask=None, past=None): argument
1025 def construct(self, input_ids, input_position, attention_mask): argument
1328 def construct(self, input_ids, input_position, attention_mask, layer_past=None, sens=None): argument
1422 input_ids, argument
|
| D | test_dynamic_pangu_alpha.py | 288 def construct(self, input_ids): argument
822 …def construct(self, input_ids, input_mask, input_position=None, attention_mask=None, layer_past=No… argument
912 def construct(self, input_ids, input_mask, input_position=None, attention_mask=None, past=None): argument
1018 def construct(self, input_ids, input_position, attention_mask): argument
1325 def construct(self, input_ids, input_position, attention_mask, layer_past=None, sens=None): argument
1418 input_ids, argument
|
| D | test_auto_parallel_frontend_scheduling_with_pangu_alpha.py | 466 def construct(self, input_ids): argument
479 def construct(self, input_ids, input_position, init_reset, batch_valid_length): argument
576 def construct(self, input_ids, argument
609 def construct(self, input_ids, input_position, attention_mask, argument
681 def construct(self, input_ids, input_position=None, attention_mask=None): argument
|
| D | test_auto_parallel_pangu_alpha.py | 467 def construct(self, input_ids): argument
480 def construct(self, input_ids, input_position, init_reset, batch_valid_length): argument
577 def construct(self, input_ids, argument
610 def construct(self, input_ids, input_position, attention_mask, argument
682 def construct(self, input_ids, input_position=None, attention_mask=None): argument
|
| D | test_layout_extend_transformer.py | 171 def construct(self, input_ids, y): argument
239 def construct(self, input_ids, y): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | device_sparse_embedding_operation.cc | 157 ParameterPtr input_ids = NewParameter(graph, kInt32, kOneDimensionalShape); in BuildEmbeddingCacheLookupKernel() local
186 ParameterPtr input_ids = NewParameter(graph, kInt32, kOneDimensionalShape); in BuildEmbeddingCacheUpdateKernel() local
214 ParameterPtr input_ids = NewParameter(graph, kInt32, kOneDimensionalShape); in BuildEmbeddingCacheEraseKernel() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/bert/src/ |
| D | bert_for_pre_training.py | 166 def construct(self, input_ids, input_mask, token_type_id, argument
244 input_ids, argument
294 input_ids, argument
384 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/mindspore-src/source/tests/ut/data/dataset/testTFBert5Rows2/ |
| D | datasetSchema.json | 5 "input_ids": { object
|
| /third_party/mindspore/mindspore-src/source/tests/ut/data/dataset/testTFBert5Rows/ |
| D | datasetSchema.json | 30 "input_ids": { object
|
| /third_party/mindspore/mindspore-src/source/tests/mindspore_test_framework/apps/ |
| D | test_bert_parts.py | 86 input_ids, input_mask, token_type_id, \ variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | vsl_compress_tensorrt.cc | 66 int *input_ids = reinterpret_cast<int *>(const_cast<void *>(inputs[C0NUM])); in enqueue() local
|
| /third_party/mindspore/mindspore-src/source/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-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sspaddmm_cpu_kernel.cc | 312 auto input_ids = static_cast<T *>(input_indices); in SparseAddSparse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/_transformer/ |
| D | transformer.py | 750 def construct(self, input_ids): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | encoder_layer_fusion.cc | 1054 position_ids, embedding_table_input, input_ids, current_index, embedding_table_pos; in CreateMaskedEncoderLayerFusionNode() local
|