Home
last modified time | relevance | path

Searched refs:kAttrOffset (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dembedding_look_up_cpu_kernel.cc76 if (AnfAlgo::HasNodeAttr(kAttrOffset, kernel_node)) { in InitKernel()
77 offset_ = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, kAttrOffset); in InitKernel()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h424 constexpr auto kAttrOffset = "offset"; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc367 emb_lookup_primitive->set_attr(kAttrOffset, MakeValue<int64_t>(0)); in CreateEmbeddingLookup()