Home
last modified time | relevance | path

Searched defs:embedding_lookup (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_0/
Dembedding_lookup.example.cpp7 namespace generated_tests::embedding_lookup { namespace
64 namespace generated_tests::embedding_lookup { namespace
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DEmbeddingLookup.cpp24 namespace embedding_lookup { namespace
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_base.py141 def embedding_lookup(self, member in TPUEmbeddingBase
H A Dtpu_embedding_for_serving.py152 def embedding_lookup(self, member in TPUEmbeddingForServing
H A Dtpu_embedding_v1.py239 def embedding_lookup(self, member in TPUEmbeddingV0
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dembedding_lookup.cc43 namespace embedding_lookup { namespace
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dembedding_ops.py263 def embedding_lookup( function
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/
H A Dtpu_embedding_v1_correctness_test.py285 def embedding_lookup(features, weights): function
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dsharded_variable.py898 def embedding_lookup(params, function
H A Dcustom_training_loop_input_test.py783 def embedding_lookup(inputs): function