Searched refs:SetNodeShardingFromNeighbors (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 219 TEST(SetNodeShardingFromNeighbors, Basic) { in TEST() argument
240 EXPECT_FALSE(SetNodeShardingFromNeighbors(c_node, /*out_edges=*/false).ok()); in TEST()
244 TF_ASSERT_OK(SetNodeShardingFromNeighbors(c_node, /*out_edges=*/false)); in TEST()
253 TF_ASSERT_OK(SetNodeShardingFromNeighbors(c_node, /*out_edges=*/false)); in TEST()
261 TF_ASSERT_OK(SetNodeShardingFromNeighbors(a_node, /*out_edges=*/true)); in TEST()
|
| D | tf2xla_util.h | 56 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges);
|
| D | functionalize_while.cc | 372 SetNodeShardingFromNeighbors(edge->dst(), /*out_edges=*/true)); in FunctionalizeLoop()
|
| D | xla_compiler.cc | 789 TF_RETURN_IF_ERROR(SetNodeShardingFromNeighbors(n, /*out_edges=*/true)); in CompileFunction()
796 TF_RETURN_IF_ERROR(SetNodeShardingFromNeighbors(n, /*out_edges=*/false)); in CompileFunction()
|
| D | tf2xla_util.cc | 500 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) { in SetNodeShardingFromNeighbors() function
|