Searched refs:field_ids (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | multifieldembeddinglookup_parallel.py | 179 param_init, target, slice_mode, sparse, operator, indices, field_ids): argument
187 self.field_ids = Tensor(field_ids)
196 x = self.embedding(self.indices, values, self.field_ids)
203 param_init, target, slice_mode, sparse, operator, indices, field_ids): argument
213 self.field_ids = field_ids
270 indices=self.indices, field_ids=self.field_ids)
278 indices=self.indices, field_ids=self.field_ids)
286 indices=self.indices, field_ids=self.field_ids)
290 indices=self.indices, field_ids=self.field_ids)
301 field_ids = np.random.randint(0, 20, (64, 64), np.int32)
[all …]
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 550 def construct(self, input_indices, input_values, field_ids): argument
554 _check_input_2d(F.shape(field_ids), "field_ids", self.cls_name)
557 _check_input_dtype(F.dtype(field_ids), "field_ids", [mstype.int32], self.cls_name)
563 field_ids = self.bias_add(field_ids, bias)
591 field_ids = self.reshape(field_ids, (-1,))
593 merged_vectors = self.merge_op(embedding, field_ids, num_segments)
596 … value_count = self.count_op(self.abs(self.reshape(input_values, (-1,))), field_ids, num_segments)
602 value_count = self.count_op(self.abs(input_values), field_ids, num_segments)
|