| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | masked_fill.cc | 31 … const BaseTensorPtr &value_tensor, const BaseTensorPtr &output_tensor) { in MaskedFillAscendCall()
39 … const BaseTensorPtr &mask_tensor, const BaseTensorPtr &value_tensor) { in MaskedFillAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ir/ |
| D | map_tensor.h | 46 TensorPtr value_tensor; member
191 const TensorPtr &value_tensor() const { return value_tensor_; } in value_tensor() function
200 void set_value_tensor(const TensorPtr value_tensor) { value_tensor_ = value_tensor; } in set_value_tensor()
|
| D | map_tensor.cc | 216 auto value_tensor = std::make_shared<Tensor>(value_dtype(), values_shape); in Export() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/experimental/ |
| D | map_parameter.py | 76 …_new__(cls, key_dtype=None, value_dtype=None, value_shape=None, key_tensor=None, value_tensor=None, argument
131 def __setitem__(self, key_tensor, value_tensor): argument
143 def _check_map_parameter_args(key_tensor, key_dtype, value_tensor, value_dtype, value_shape): argument
227 def put(self, key_tensor, value_tensor): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/control/ |
| D | tensorlist_setitem_infer.c | 22 int PreJudge(const TensorC *get_index, TensorListC *input0, const TensorC *value_tensor) { in PreJudge()
48 const TensorC *value_tensor = inputs[2]; in TensorListSetItemInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | fill.cc | 63 Tensor *value_tensor = ctx.Input(1); in Compute() local
142 void FillCpuKernel::FillOutput(CpuKernelContext &ctx, const Tensor *value_tensor, Tensor *output) { in FillOutput()
|
| D | masked_fill.cc | 48 Tensor *value_tensor = ctx.Input(kThirdInputIndex); in GetValueFromVariousTypesMaskedFill() local
|
| D | ragged_tensor_to_tensor.cc | 553 const auto value_tensor = ctx.Input(kValueInputIndex); in SetOutput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | constantofshape_tensorrt.cc | 53 nvinfer1::ITensor *value_tensor; in AddInnerOp() local
|
| D | oneslike_tensorrt.cc | 55 nvinfer1::ITensor *value_tensor; in RunAsTrtOps() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/environ/ |
| D | environ_set.cc | 80 auto *value_tensor = ctx.Input(kThirdInputIndex); in ParseKernelParam() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_neg.cc | 55 auto value_tensor = value->cast<tensor::TensorPtr>(); in TEST_P() local
|
| D | test_ops_notequal.cc | 91 auto value_tensor = value->cast<tensor::TensorPtr>(); in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | one_hot.c | 84 TensorC *value_tensor = one_hot->base_.in_[THIRD_INPUT]; in OneHotInitOnOffValueForThreeInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | _ps_context.py | 241 def _store_warm_up_ptr_by_tensor_list(param_key, key_tensor, value_tensor, status_tensor): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/ |
| D | setitem_impl.py | 277 def _tensor_setitem_by_tensor_with_tensor(data, index, value_tensor): argument
864 def _map_tensor_setitem(map_tensor, key_tensor, value_tensor): argument
|
| D | _compile_utils.py | 986 def tensor_setitem_by_tensor_with_tensor(data, index, value_tensor): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | map_tensor_py.cc | 145 const py::object &permit_filter_obj, const py::object &evict_filter_obj) { in RegMapTensor()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_auto_parallel_frontend_scheduling_with_pangu_alpha.py | 268 def construct(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
302 def _convert_to_2d_tensor(self, query_tensor, key_tensor, value_tensor, attention_mask): argument
|
| D | test_auto_parallel_pangu_alpha.py | 269 def construct(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
303 def _convert_to_2d_tensor(self, query_tensor, key_tensor, value_tensor, attention_mask): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/_transformer/ |
| D | transformer.py | 1127 def construct(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
1233 def _check_inputs(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
1261 def _convert_to_2d_tensor(self, query_tensor, key_tensor, value_tensor, attention_mask): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/proto/ |
| D | mind_ir.proto | 221 required TensorProto value_tensor = 4; field
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | data_utils.cc | 551 const std::shared_ptr<Tensor> &value_tensor, RelationalOp op) { in MaskHelper()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/expander/core/ |
| D | emitter.h | 233 auto value_tensor = Cast(Tensor(value), data_type); in Fill() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runner.cc | 235 auto value_tensor = map_tensor->value_tensor(); in CopyMapTensorDataToDevice() local
|