| /third_party/mindspore/mindspore/parallel/nn/ |
| D | loss.py | 94 def construct(self, logits, label, input_mask): argument
127 def _check_input(self, logits, label, input_mask): argument
|
| D | transformer.py | 447 def construct(self, input_mask): argument
1239 def construct(self, x, input_mask, init_reset=True, batch_valid_length=None): argument
1312 def _check_input(self, x, input_mask, init_reset, batch_valid_length): argument
|
| /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 | 709 def construct(self, input_mask): 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
244 input_mask, argument
294 input_mask, argument
383 input_mask, argument
|
| D | bert_model.py | 824 def construct(self, input_mask): argument
915 def construct(self, input_ids, token_type_ids, input_mask): argument
|
| /third_party/mindspore/tests/ut/data/dataset/testTFBert5Rows1/ |
| D | datasetSchema.json | 25 "input_mask": { object
|
| /third_party/mindspore/tests/ut/data/dataset/testTFBert5Rows/ |
| D | datasetSchema.json | 25 "input_mask": { object
|
| /third_party/libinput/include/linux/linux/ |
| D | input.h | 116 struct input_mask { struct
117 __u32 type;
118 __u32 codes_size;
119 __u64 codes_ptr;
|
| /third_party/libevdev/include/linux/linux/ |
| D | input.h | 121 struct input_mask { struct
122 __u32 type;
123 __u32 codes_size;
124 __u64 codes_ptr;
|
| /third_party/libinput/include/linux/freebsd/ |
| D | input.h | 117 struct input_mask { struct
118 uint32_t type;
119 uint32_t codes_size;
120 uint64_t codes_ptr;
|
| /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 | 160 input_mask = M.Tensor(np.zeros((32, 128), np.int32)) variable
|
| /third_party/mesa3d/src/gallium/drivers/llvmpipe/ |
| D | lp_linear.c | 95 unsigned input_mask = variant->linear_input_mask; in lp_fs_linear_run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | pooling_grad_gpu_kernel.h | 88 auto input_mask = AnfAlgo::GetInputDeviceShape(kernel_node, 1); in InitShape() local
|
| /third_party/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_shader.c | 1323 uint16_t input_mask = u_bit_consecutive(0, info->num_inputs); in si_get_vs_prolog_key() local
|
| /third_party/mesa3d/src/amd/common/ |
| D | ac_nir_lower_ngg.c | 220 nir_ssa_def *input_mask = nir_build_ballot(b, 1, wave_size, input_bool); in repack_invocations_in_workgroup() local
|
| /third_party/mesa3d/src/amd/vulkan/ |
| D | radv_shader.h | 307 uint32_t input_mask; member
|