Home
last modified time | relevance | path

Searched refs:GetEnqueueOpsFromGraph (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.cc176 Status UpdateTPUEmbeddingModePass::GetEnqueueOpsFromGraph( in GetEnqueueOpsFromGraph() function in tensorflow::UpdateTPUEmbeddingModePass
303 TF_RETURN_IF_ERROR(GetEnqueueOpsFromGraph(graph, &enqueue_nodes)); in Run()
H A Dupdate_tpu_embedding_ops_passes.h37 static Status GetEnqueueOpsFromGraph(