Searched refs:TPUEmbedding (Results 1 – 18 of 18) sorted by relevance
62 struct TPUEmbedding : ::mlir::SideEffects::Resource::Base<TPUEmbedding> { struct
151 def TF_TPUEmbeddingResource : TF_ResourceBase<"TPUEmbedding">;
2086 ResourceEffects::TPUEmbedding::get()); in getEffects()2152 ResourceEffects::TPUEmbedding::get()); in getEffects()
907 An op that enqueues a list of input batch tensors to TPUEmbedding.
2108 let summary = "Sets up TPUEmbedding in a distributed TPU system.";3721 An op that enqueues a list of input batch tensors to TPUEmbedding.3787 An op that enqueues TPUEmbedding input indices from a SparseTensor.
1 path: "tensorflow.tpu.experimental.embedding.TPUEmbedding"3 is_instance: "<class \'tensorflow.python.tpu.tpu_embedding_v2.TPUEmbedding\'>"
20 name: "TPUEmbedding"
11 summary: "Sets up TPUEmbedding in a distributed TPU system."
27 summary: "An op that enqueues a list of input batch tensors to TPUEmbedding."
54 summary: "An op that enqueues TPUEmbedding input indices from a SparseTensor."
106 return tpu_embedding_v2.TPUEmbedding(295 mid_level = tpu_embedding_v2.TPUEmbedding(320 mid_level = tpu_embedding_v2.TPUEmbedding(345 mid_level = tpu_embedding_v2.TPUEmbedding(
180 self.tpu_embedding = tpu_embedding_v2.TPUEmbedding(233 mid_level = tpu_embedding_v2.TPUEmbedding(404 tpu_embedding_v2.TPUEmbedding(425 tpu_embedding_v2.TPUEmbedding(1038 return tpu_embedding_v2.TPUEmbedding(1151 mid_level_api = tpu_embedding_v2.TPUEmbedding(1227 mid_level = tpu_embedding_v2.TPUEmbedding(1267 mid_level_api = tpu_embedding_v2.TPUEmbedding(1294 mid_level_api = tpu_embedding_v2.TPUEmbedding(
214 return tpu_embedding_v2.TPUEmbedding(489 mid_level = tpu_embedding_v2.TPUEmbedding(
96 class TPUEmbedding(tracking.AutoTrackable): class
1105 class TPUEmbedding(object): class
99 if (llvm::isa<ResourceEffects::TPUEmbedding>(effect.getResource())) { in MustExecute()