Searched defs:value_tensor (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/control/ |
| D | tensorlist_setitem_infer.c | 20 int PreJudge(const TensorC *get_index, TensorListC *input0, const TensorC *value_tensor) { in PreJudge()
46 const TensorC *value_tensor = inputs[2]; in TensorListSetItemInferShape() local
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 804 def construct(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
885 def _check_inputs(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
938 def _convert_to_2d_tensor(self, query_tensor, key_tensor, value_tensor, attention_mask): argument
|
| /third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
| D | setitem_impl.py | 154 def _tensor_setitem_by_tensor_with_tensor(data, index, value_tensor): argument
|
| D | _compile_utils.py | 657 def tensor_setitem_by_tensor_with_tensor(data, index, value_tensor): argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | data_utils.cc | 508 const std::shared_ptr<Tensor> &value_tensor, RelationalOp op) { in MaskHelper()
|