| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/ |
| D | fused_infer_attention_score_aclnn_kernel.cc | 66 auto inner_precise = transform::ConvertKernelTensor<int64_t>(inputs[real_input_idx_[kIndex24]]); in GetWorkSpaceInfo() local
120 auto inner_precise = transform::ConvertKernelTensor<int64_t>(inputs[real_input_idx_[kIndex24]]); in Launch() local
|
| D | flash_attention_score_grad_aclnn_kernel.h | 87 auto inner_precise = inputs[kIndex20]; in DEFINE_GET_WORKSPACE_FOR_RESIZE() local
|
| D | flash_attention_score_aclnn_kernel.cc | 58 auto inner_precise = inputs[kIndex15]; in GetWorkSpaceInfo() local
|
| D | flash_attention_score_grad_aclnn_kernel.cc | 58 auto inner_precise = inputs[kIndex20]; in GetWorkSpaceInfo() local
|
| D | flash_attention_score_aclnn_kernel.h | 87 auto inner_precise = inputs[kIndex15]; in DEFINE_GET_WORKSPACE_FOR_RESIZE() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_fused_infer_attention_score.py | 33 inner_precise=0): argument
53 … antiquant_scale, antiquant_offset, block_table=None, block_size=0, inner_precise=0): argument
63 pre_tokens=2147483547, next_tokens=0, sparse_mode=0, inner_precise=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | flash_attention_fusion.h | 32 int inner_precise = 1; member
|
| D | flash_attention_fusion.cc | 2656 int inner_precise = std::atoi(attr_value.c_str()); in ParseFAParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | flash_attention_score.cc | 50 const Int64ImmPtr next_tokens, const Int64ImmPtr inner_precise, const Int64ImmPtr input_layout, in FlashAttentionScoreAscendCall()
105 const Int64ImmPtr inner_precise, const Int64ImmPtr input_layout, const Int64ImmPtr sparse_mode) { in FlashAttentionScoreAscendCustomize()
|
| D | flash_attention_score_grad.cc | 39 const Int64ImmPtr inner_precise, const Int64ImmPtr input_layout, const Int64ImmPtr sparse_mode, in FlashAttentionScoreGradAscendCall()
93 const Int64ImmPtr inner_precise, const Int64ImmPtr input_layout, const Int64ImmPtr sparse_mode) { in FlashAttentionScoreGradAscendCustomize()
|
| D | ffn_ext.cc | 37 const Int64ImmPtr &activation, const Int64ImmPtr &inner_precise) { in FFNExtAscendCustomize()
|
| D | fused_infer_attention_score.cc | 40 …Int64ImmPtr &num_key_value_heads, const Int64ImmPtr &sparse_mode, const Int64ImmPtr &inner_precise, in FusedInferAttentionScoreAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_prompt_flash_attention_parser.cc | 52 int64_t inner_precise = onnx_node_attr.i(); in Parse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | f_f_n.cc | 75 void FFN::Init(const std::string &activation, int64_t inner_precise) { in Init()
84 void FFN::set_inner_precise(int64_t inner_precise) { in set_inner_precise()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_flash_attention_score.py | 31 … next_tokens=65536, scalar_value=1.0, inner_precise=0, sparse_mode=0): argument
95 next_tokens=65536, inner_precise=0, sparse_mode=0): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_fused_infer_attention_score.py | 152 num_key_value_heads=0, strategy=None, sparse_mode=0, inner_precise=1, block_size=0, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/ |
| D | gen_lite_ops.cc | 728 void FFNExt::set_inner_precise(const int64_t &inner_precise) { (void)this->AddAttr("inn… in set_inner_precise()
803 void FlashAttentionScoreGrad::set_inner_precise(const int64_t &inner_precise) { (void)t… in set_inner_precise()
838 void FlashAttentionScore::set_inner_precise(const int64_t &inner_precise) { (void)this-… in set_inner_precise()
2200 void FusedInferAttentionScore::set_inner_precise(const int64_t &inner_precise) { (void)… in set_inner_precise()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | nn_ops.py | 9606 num_key_value_heads=0, sparse_mode=0, inner_precise=1): argument
9675 inner_precise=1): argument
|
| D | _inner_ops.py | 2469 def __init__(self, activation, inner_precise): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | nn_func.py | 7889 num_key_value_heads=0, sparse_mode=0, inner_precise=1): argument
7965 inner_precise=1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/manually_defined/ |
| D | ops_def.py | 1873 … scalar_value=1.0, pre_tokens=2147483647, next_tokens=2147483647, inner_precise=0, argument
|