Home
last modified time | relevance | path

Searched refs:UpdateGraphEnqueueOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dupdate_tpu_embedding_ops_passes.cc197 Status UpdateTPUEmbeddingModePass::UpdateGraphEnqueueOp(bool training, in UpdateGraphEnqueueOp() function in tensorflow::UpdateTPUEmbeddingModePass
306 UpdateGraphEnqueueOp(enqueue.second, graph, enqueue.first)); in Run()
H A Dupdate_tpu_embedding_ops_passes.h39 static Status UpdateGraphEnqueueOp(bool training, Graph* graph,