Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.h172 const NodeDef* enqueue_node,
H A Dgraph_properties.cc2469 const NodeDef* enqueue_node, in UpdateEnqueue() argument
2472 auto ctx = shape_refiner->GetNodeContext(enqueue_node); in UpdateEnqueue()
2474 TF_RETURN_IF_ERROR(shape_refiner->AddNode(enqueue_node)); in UpdateEnqueue()
2475 ctx = CHECK_NOTNULL(shape_refiner->GetNodeContext(enqueue_node)); in UpdateEnqueue()
2478 auto it = resource_handles.find(enqueue_node); in UpdateEnqueue()
2493 GraphView::InputPort inp(enqueue_node, i); in UpdateEnqueue()