Home
last modified time | relevance | path

Searched defs:embedding_layer (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dtpu_embedding_software_deduplication_rewrite_pass.cc139 std::string embedding_layer; in MakeRecvActivationsNodeDef() local
231 std::string embedding_layer; in MakeSendGradientsNodeDef() local
/aosp_15_r20/external/executorch/examples/mediatek/model_export_scripts/
H A Dllama.py228 embedding_layer, argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py906 embedding_layer, # torch.nn.Embedding. Must have a padding_idx field argument