Searched refs:mask_input (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | dropout_unify_mindir.cc | 380 auto mask_input = dropout_grad_cnode->input(kIndex2); in Process() local
381 if (mask_input->isa<Parameter>()) { in Process()
383 auto mask_abstract = mask_input->abstract(); in Process()
387 mask_input->set_abstract(mask_abstract); in Process()
392 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), mask_input.get()); in Process()
398 grad_input, mask_input, keep_prob_value}; in Process()
|
| /third_party/mesa3d/src/gallium/drivers/llvmpipe/ |
| D | lp_state_fs.c | 226 LLVMValueRef mask_input) /* int64 */ in generate_quad_mask() argument
265 mask_input = LLVMBuildLShr(builder, mask_input, lp_build_const_int64(gallivm, 16 * sample), ""); in generate_quad_mask()
266 mask_input = LLVMBuildTrunc(builder, mask_input, in generate_quad_mask()
268 mask_input = LLVMBuildAnd(builder, mask_input, lp_build_const_int32(gallivm, 0xffff), ""); in generate_quad_mask()
270 mask_input = LLVMBuildLShr(builder, in generate_quad_mask()
271 mask_input, in generate_quad_mask()
280 mask_input); in generate_quad_mask()
3013 LLVMValueRef mask_input; in generate_fragment() local
3122 mask_input = LLVMGetParam(function, 9); in generate_fragment()
3137 lp_build_name(mask_input, "mask_input"); in generate_fragment()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | multi_head_attention_fusion.cc | 45 VectorRef DefineMask(const BaseRef &mask_input) { in DefineMask() argument
50 auto expand_dims = VectorRef({is_expand_dims, mask_input, is_param1}); in DefineMask()
|