Home
last modified time | relevance | path

Searched refs:kStreaming (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc44 const char kStreaming[] = "_streaming"; variable
114 if (!node.attr().contains(kStreaming)) return false; in IsStreamingPort()
116 auto& attr_list = node.attr().at(kStreaming).list(); in IsStreamingPort()
715 if (from->attr().contains(kStreaming) && !control_input) { in CreateSendRecv()
716 if (input_node_port_num >= from->attr().at(kStreaming).list().b_size()) { in CreateSendRecv()
720 } else if (from->attr().at(kStreaming).list().b(input_node_port_num)) { in CreateSendRecv()
721 send_attr[kStreaming].mutable_list()->add_b(true); in CreateSendRecv()
722 recv_attr[kStreaming].mutable_list()->add_b(true); in CreateSendRecv()
Dvirtual_scheduler.h40 ABSL_CONST_INIT extern const char kStreaming[];
Dvirtual_scheduler_test.cc2550 (*node.mutable_attr())[kStreaming].mutable_list()->add_b(true); in TEST_F()