Searched defs:input_v (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_inplace_op.py | 39 def construct(self, x, input_v): argument
48 def construct(self, x, input_v): argument
57 def construct(self, x, input_v): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | inplace_op_impl.cu | 36 __global__ void InplaceUpdate(const size_t size, const T *input_v, T *output, const S *indices, S *… in InplaceUpdate()
53 __global__ void InplaceAddOrSub(const size_t size, const T *input_v, T *output, const S *indices, in InplaceAddOrSub()
66 cudaError_t CalInplaceOp(const size_t size_v, const T *input_v, T *output, S *indices, S *indices_k… in CalInplaceOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/ |
| D | dsd_impl.py | 37 def dsd_matmul(input_w1, input_w2, input_v, output_y={}, kernel_name='dsd_matmul'): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/pi_jit/share/ops/primitive/ |
| D | inplaceadd_ops.py | 12 def construct(self, input_x, input_v): argument
|
| D | inplacesub_ops.py | 14 def construct(self, input_x, input_v): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | inplace_op_gpu_kernel.cc | 103 T *input_v = GetDeviceAddress<T>(inputs, kIndex1); in LaunchKernel() local
|
| D | inplace_op_v2_gpu_kernel.cc | 97 T *input_v = GetDeviceAddress<T>(inputs, kIndex2); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 236 auto input_v = utils::cast<AnfNodePtr>((*equiv)[input_v_]); in CreateRelPosMultiHeadAttentionNode() local
|
| D | multi_head_attention_fusion.cc | 929 auto input_v = utils::cast<AnfNodePtr>((*equiv)[input_v_]); in IsCross() local
966 auto input_v = utils::cast<AnfNodePtr>((*equiv)[input_v_]); in GetNewNodeInputs() local
|
| D | flash_attention_fusion.cc | 634 auto input_v = std::make_shared<Var>(); in DefineFlashAttentionPatternForVideoComposer() local
|