Home
last modified time | relevance | path

Searched refs:EmbeddingLookup (Results 1 – 25 of 42) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dembedding_lookup_gpu_kernel.cc22 EmbeddingLookup,
26 EmbeddingLookup,
30 EmbeddingLookup,
34 EmbeddingLookup,
38 EmbeddingLookup,
42 EmbeddingLookup,
46 EmbeddingLookup,
50 EmbeddingLookup,
54 EmbeddingLookup,
58 EmbeddingLookup,
[all …]
/third_party/mindspore/mindspore/core/ops/
Dembedding_lookup.h32 class MS_CORE_API EmbeddingLookup : public PrimitiveC {
35 …EmbeddingLookup() : PrimitiveC(kNameEmbeddingLookup) { InitIOName({"params", "indices", "offset"},… in EmbeddingLookup() function
37 ~EmbeddingLookup() = default;
38 MS_DECLARE_PARENT(EmbeddingLookup, PrimitiveC);
50 using PrimEmbeddingLookupPtr = std::shared_ptr<EmbeddingLookup>;
Dembedding_lookup.cc25 void EmbeddingLookup::Init(const bool setattr_flag) { this->set_setattr_flag(setattr_flag); } in Init()
27 void EmbeddingLookup::set_setattr_flag(const bool setattr_flag) { in set_setattr_flag()
31 bool EmbeddingLookup::get_setattr_flag() const { in get_setattr_flag()
80 REGISTER_PRIMITIVE_C(kNameEmbeddingLookup, EmbeddingLookup);
/third_party/mindspore/tests/mindspore_test_framework/apps/
Dtest_bert_parts.py21 from mindspore.tests.models.Bert_NEZHA import EmbeddingLookup, GetMaskedLMOutput, \
133 'block': get_output_cell(EmbeddingLookup(vocab_size=32000,
141 'block': get_output_cell(EmbeddingLookup(vocab_size=32000,
149 'block': get_output_cell(EmbeddingLookup(vocab_size=32000,
157 'block': get_output_cell(EmbeddingLookup(vocab_size=32000,
165 'block': EmbeddingLookup(vocab_size=32000,
173 'block': (get_output_cell(EmbeddingLookup(vocab_size=32000,
182 'block': (EmbeddingLookup(vocab_size=32000,
191 'block': (EmbeddingLookup(vocab_size=32000,
Dtest_bert_compare_with_npy.py23 EmbeddingLookup, BertModel, \
467 'block': EmbeddingLookup(vocab_size=21128,
Dtest_bert_check_gradient.py24 EmbeddingLookup, EmbeddingPostprocessor, BertOutput, \
286 'block': EmbeddingLookup(vocab_size=21128,
/third_party/mindspore/tests/ut/python/parallel/
Dtest_multi_field_embedding.py59 …def __init__(self, shape, field_size=10, slice_mode=nn.EmbeddingLookup.BATCH_SLICE, target="Device…
95 …net = NetWithLoss(Net(shape, field_size=9, slice_mode=nn.EmbeddingLookup.TABLE_ROW_SLICE, target='…
102 …net = NetWithLoss(Net(shape, field_size=10, slice_mode=nn.EmbeddingLookup.TABLE_COLUMN_SLICE, targ…
116 …net = NetWithLoss(Net(shape, target='DEVICE', slice_mode=nn.EmbeddingLookup.TABLE_COLUMN_SLICE, op…
123 …net = NetWithLoss(Net(shape, target='DEVICE', slice_mode=nn.EmbeddingLookup.TABLE_ROW_SLICE, opera…
137 …net = NetWithLoss(Net(shape, target='DEVICE', slice_mode=nn.EmbeddingLookup.TABLE_COLUMN_SLICE, op…
144 …net = NetWithLoss(Net(shape, target='DEVICE', slice_mode=nn.EmbeddingLookup.TABLE_ROW_SLICE, opera…
Dtest_sparse_feature_bprop.py84 self.embeddinglookup = nn.EmbeddingLookup(64, 64, param_init='ones')
Dtest_embeddinglookup.py52 self.elu = P.EmbeddingLookup().shard(strategy1).add_prim_attr("primitive_target", target)
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dembedding_look_up_cpu_kernel.h50 EmbeddingLookup,
55 EmbeddingLookup,
60 EmbeddingLookup,
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_embeddinglookup.py45 class EmbeddingLookup(Cell): class
48 self.op = op.EmbeddingLookup()
64 net = EmbeddingLookup(self.offset)
/third_party/mindspore/mindspore/train/train_thor/
Dconvert_utils.py31 nn.EmbeddingLookup: ConvertNetUtils._convert_embeddinglookup}
119 …if isinstance(subcell, (nn.Embedding, nn.EmbeddingLookup)) and subcell.embedding_table.requires_gr…
138 elif isinstance(subcell, (nn.Embedding, nn.Dense, nn.Conv2d, nn.EmbeddingLookup)):
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dembedding_lookup_fp32.h37 int EmbeddingLookup(float *input_data, const int *ids, float *output_data, const EmbeddingLookupPar…
Dembedding_lookup_fp32.c50 int EmbeddingLookup(float *input_data, const int *ids, float *output_data, const EmbeddingLookupPar… in EmbeddingLookup() function
/third_party/mindspore/tests/st/networks/models/bert/src/
D__init__.py20 BertOutput, BertSelfAttention, BertTransformer, EmbeddingLookup, \
/third_party/mindspore/tests/st/ops/ascend/
Dtest_embedding_lookup.py30 self.embedding = P.EmbeddingLookup()
/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_embeddinglookup.cc35 auto embedding_lookup = std::make_shared<EmbeddingLookup>(); in TEST_F()
/third_party/mindspore/tests/st/ops/gpu/
Dtest_embedding_lookup_op.py28 self.embeddinglookup = P.EmbeddingLookup()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dembedding_lookup_infer.c77 REG_INFER(EmbeddingLookup, PrimType_EmbeddingLookupFusion, EmbeddingLookupInferShape)
/third_party/mindspore/tests/st/dynamic_shape/
Dtest_dynamic_shape_embedding.py30 nn.EmbeddingLookup(vocab_size=vocab_size,
/third_party/mindspore/mindspore/nn/layer/
Dembedding.py156 class EmbeddingLookup(Cell): class
227 super(EmbeddingLookup, self).__init__()
243 self.embeddinglookup = P.EmbeddingLookup().add_prim_attr('primitive_target', 'CPU')
401 class MultiFieldEmbeddingLookup(EmbeddingLookup):
/third_party/mindspore/tests/st/ps/part_ps/
Dtest_ps_embedding_heterogeneous_conv2d_adam.py32 from mindspore.nn import EmbeddingLookup
59 self.embedding_lookup = EmbeddingLookup(vocab_size=vocab_size,
/third_party/mindspore/tests/st/ps/cmp_sparse_embedding/
Dtest_cmp_sparse_embedding.py46 self.embedding = nn.EmbeddingLookup(16, 4)
/third_party/mindspore/tests/st/ops/cpu/
Dtest_embedding_look_up_op.py29 self.embedding = P.EmbeddingLookup().add_prim_attr("primitive_target", "CPU")
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dembedding_lookup_fp32.cc58 int error_code = EmbeddingLookup(input_addr_, ids_addr, output_addr, param_, task_id); in DoExcute()

12