Searched refs:key_placeholder (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 109 const std::vector<DataType>& recv_at_host_dtypes, Node* key_placeholder) { in BuildRecvAtHostNode() argument
123 recv_at_host_builder.Input(key_placeholder->name(), 0, DT_STRING); in BuildRecvAtHostNode()
134 std::vector<DataType>* recv_at_host_dtypes, Node* key_placeholder) { in ReplaceArgNodesWithRecvAtHostNode() argument
142 key_placeholder)); in ReplaceArgNodesWithRecvAtHostNode()
184 g->AddEdge(key_placeholder, 0, recv_at_host_node, 0); in ReplaceArgNodesWithRecvAtHostNode()
211 const std::vector<DataType>& send_from_host_dtypes, Node* key_placeholder) { in BuildSendFromHostNode() argument
240 send_from_host_builder.Input(key_placeholder->name(), 0, DT_STRING); in BuildSendFromHostNode()
251 std::vector<DataType>* send_from_host_dtypes, Node* key_placeholder) { in ReplaceRetNodesWithSendFromHostNode() argument
259 *send_from_host_dtypes, key_placeholder)); in ReplaceRetNodesWithSendFromHostNode()
272 g->AddEdge(key_placeholder, 0, send_from_host_node, in ReplaceRetNodesWithSendFromHostNode()
[all …]
|
| D | extract_outside_compilation_pass_test.cc | 69 Node *key_placeholder = node_name_image["cluster_key_placeholder"]; in TEST() local
70 EXPECT_NE(key_placeholder, nullptr); in TEST()
140 Node *key_placeholder = node_name_image["cluster_key_placeholder"]; in TEST() local
141 EXPECT_NE(key_placeholder, nullptr); in TEST()
166 Node *key_placeholder = node_name_image["cluster_key_placeholder"]; in TEST() local
167 EXPECT_NE(key_placeholder, nullptr); in TEST()
190 Node *key_placeholder = node_name_image["cluster_key_placeholder"]; in TEST() local
191 EXPECT_EQ(key_placeholder, nullptr); in TEST()
350 Node *key_placeholder = nullptr, *sequencer = nullptr; in TEST_F() local
354 EXPECT_EQ(key_placeholder, nullptr); in TEST_F()
[all …]
|