Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc581 if (input_node->attr().count(kAttrTensorName)) { in CreateSendRecv()
582 send_attr[kAttrTensorName].set_s( in CreateSendRecv()
583 input_node->attr().at(kAttrTensorName).s()); in CreateSendRecv()
594 if (input_node->attr().count(kAttrTensorName)) { in CreateSendRecv()
595 recv_attr[kAttrTensorName].set_s( in CreateSendRecv()
596 input_node->attr().at(kAttrTensorName).s()); in CreateSendRecv()
Dvirtual_scheduler.h308 const string kAttrTensorName = "tensor_name"; variable