Searched refs:UpdateNodeDevice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view.h | 291 void UpdateNodeDevice(MutableNodeView* node, absl::string_view device);
294 void UpdateNodeDevice(const MutationNewNode& node, absl::string_view device);
|
| D | graph_view_test.cc | 1139 mutation->UpdateNodeDevice(a_node, "foo"); in TEST_F()
1398 mutation->UpdateNodeDevice(d_node, kDeviceGPU0); in TEST_F()
1626 mutation->UpdateNodeDevice(new_node, kDeviceGPU2); in TEST_F()
1666 mutation->UpdateNodeDevice(d_node, kDeviceGPU2); in TEST_F()
1872 mutation->UpdateNodeDevice(new_node_2_node, kDeviceGPU3); in TEST_F()
1915 mutation->UpdateNodeDevice(a_node, kDeviceGPU1); in TEST_F()
|
| D | graph_view.cc | 318 void Mutation::UpdateNodeDevice(MutableNodeView* node, in UpdateNodeDevice() function in tensorflow::grappler::utils::Mutation
325 void Mutation::UpdateNodeDevice(const MutationNewNode& node, in UpdateNodeDevice() function in tensorflow::grappler::utils::Mutation
|