Searched refs:SetNodeShardingFromNeighbors (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 217 TEST(SetNodeShardingFromNeighbors, Basic) { in TEST() argument
238 EXPECT_FALSE(SetNodeShardingFromNeighbors(c_node, /*out_edges=*/false).ok()); in TEST()
242 TF_ASSERT_OK(SetNodeShardingFromNeighbors(c_node, /*out_edges=*/false)); in TEST()
250 TF_ASSERT_OK(SetNodeShardingFromNeighbors(c_node, /*out_edges=*/false)); in TEST()
257 TF_ASSERT_OK(SetNodeShardingFromNeighbors(a_node, /*out_edges=*/true)); in TEST()
|
| D | tf2xla_util.h | 55 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges);
|
| D | xla_compiler.cc | 585 TF_RETURN_IF_ERROR(SetNodeShardingFromNeighbors(n, /*out_edges=*/true)); in CompileFunction()
592 TF_RETURN_IF_ERROR(SetNodeShardingFromNeighbors(n, /*out_edges=*/false)); in CompileFunction()
|
| D | functionalize_while.cc | 448 SetNodeShardingFromNeighbors(edge->dst(), /*out_edges=*/true)); in FunctionalizeLoop()
|
| D | tf2xla_util.cc | 500 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) { in SetNodeShardingFromNeighbors() function
|