| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_embedding_lookup_op.py | 30 def construct(self, input_params, input_indices, offset): argument
31 return self.embeddinglookup(input_params, input_indices, offset)
37 input_indices = Tensor(np.array([[5, 2], [8, 5]]).astype(np.int32))
39 output = Net()(input_params, input_indices, offset)
45 input_indices = Tensor(np.array([[5, 2], [8, 5]]).astype(np.int32))
47 output = Net()(input_params, input_indices, offset)
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | input_to_output_registry.h | 35 …void set_input_indices(const std::vector<size_t> &input_indices) { input_indices_ = input_indices;… in set_input_indices() argument
37 const std::vector<size_t> &input_indices() const { return input_indices_; } in input_indices() function
51 const std::string &op_name, const std::vector<size_t> &input_indices,
|
| D | add_input_to_output.cc | 38 void AddOutputs(const CNodePtr &cnode, const std::vector<size_t> &input_indices) { in AddOutputs() argument
56 for (size_t i = 0; i < input_indices.size(); ++i) { in AddOutputs()
57 size_t index = input_indices[i]; in AddOutputs()
100 AddOutputs(cnode, reg.input_indices()); in Process()
|
| D | input_to_output_registry.cc | 103 …putToOutputRegistry::Register(const std::string &op_name, const std::vector<size_t> &input_indices, in Register() argument
107 reg.set_input_indices(input_indices); in Register()
|
| /third_party/mindspore/mindspore/lite/micro/coder/ |
| D | graph.cc | 132 std::vector<uint32_t> input_indices; in InitGraphInOutTensors() local
142 input_indices.push_back(static_cast<uint32_t>(in_tensor_index)); in InitGraphInOutTensors()
156 SetInputIndices(input_indices); in InitGraphInOutTensors()
216 void CoderGraph::SetInputIndices(const std::vector<uint32_t> &input_indices) { in SetInputIndices() argument
217 input_indices_.insert(input_indices_.end(), input_indices.begin(), input_indices.end()); in SetInputIndices()
244 std::vector<uint32_t> CoderGraph::input_indices() const { return this->input_indices_; } in input_indices() function in mindspore::lite::micro::CoderGraph
|
| D | session.cc | 251 std::vector<uint32_t> input_indices; in CreateOpCoders() local
253 input_indices.insert(input_indices.end(), node_input_indices.begin(), node_input_indices.end()); in CreateOpCoders()
260 for (auto in_index : input_indices) { in CreateOpCoders()
302 .input_indices(input_indices) in CreateOpCoders()
|
| D | graph.h | 46 void SetInputIndices(const std::vector<uint32_t> &input_indices);
54 std::vector<uint32_t> input_indices() const;
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/ |
| D | op_coder.cc | 34 void OperatorCoder::set_input_tensor_indices(const std::vector<uint32_t> &input_indices) { in set_input_tensor_indices() argument
35 input_tensor_indices_ = input_indices; in set_input_tensor_indices()
|
| D | op_coder_builder.h | 42 OpCoderBuilder &input_indices(const std::vector<uint32_t> &indices);
|
| D | op_coder_builder.cc | 92 OpCoderBuilder &OpCoderBuilder::input_indices(const std::vector<uint32_t> &indices) { in input_indices() function in mindspore::lite::micro::OpCoderBuilder
|
| D | op_coder.h | 52 void set_input_tensor_indices(const std::vector<uint32_t> &input_indices);
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 550 def construct(self, input_indices, input_values, field_ids): argument
552 _check_input_2d(F.shape(input_indices), "input_indices", self.cls_name)
555 …_check_input_dtype(F.dtype(input_indices), "input_indices", [mstype.int32, mstype.int64], self.cls…
559 batch_size = self.shape(input_indices)[0]
566 out = self.embeddinglookup(self.embedding_table, input_indices, 0)
569 shp = self.shape(input_indices) + (self.embedding_size,)
570 indices_flatten = self.reshape(input_indices, (-1,))
576 out = self.gatherv2(self.embedding_table, input_indices, 0)
578 axis = _make_axis_range(F.rank(input_indices), F.rank(out))
582 weights = self.reshape(input_values, (batch_size, self.shape(input_indices)[1], 1))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | map_cache_idx_cpu_kernel.cc | 103 auto input_indices = reinterpret_cast<T *>(inputs[1]->addr); in LaunchKernel() local
118 T key = input_indices[i] - offset; in LaunchKernel()
|
| /third_party/mindspore/ |
| D | RELEASE.md | 2230 >>> input_indices = Tensor(np.array([[1, 0], [3, 2]]), mindspore.int32)
2231 >>> result = EmbeddingLookup(4,2)(input_indices)
2242 >>> input_indices = Tensor(np.array([[1, 0], [3, 2]]), mindspore.int32)
2243 >>> result = EmbeddingLookup(4,2)(input_indices, sparse=False)
|