Searched refs:schema_dir (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | dataset.py | 26 …ert_dataset(epoch_size=1, device_num=1, rank=0, do_shuffle="true", data_dir=None, schema_dir=None): argument
35 data_set = ds.TFRecordDataset(data_files, schema_dir if schema_dir != "" else None,
|
| /third_party/mindspore/cmake/external_libs/ |
| D | flatbuffers.cmake | 52 foreach(schema_dir ${source_schema_dirs})
53 set(total_schema_dirs -I ${schema_dir} ${total_schema_dirs})
90 foreach(schema_dir ${source_schema_dirs})
91 set(total_schema_dirs -I ${schema_dir} ${total_schema_dirs})
|
| /third_party/mindspore/tests/st/networks/models/bert/bert_performance/ |
| D | test_bert_thor.py | 108 def create_bert_dataset(device_num=1, rank=0, do_shuffle="true", data_dir=None, schema_dir=None): argument
117 data_set = ds.TFRecordDataset(data_files, schema_dir if schema_dir != "" else None,
161 schema_dir=None)
|