Searched refs:leaf_count (Results 1 – 10 of 10) sorted by relevance
52 const int64 leaf_count = output_buffers->leaf_count(); in ExecuteOnStream() local53 TF_RET_CHECK(source_slices_.size() == leaf_count) in ExecuteOnStream()55 << ") and outputs (" << leaf_count << ")"; in ExecuteOnStream()58 for (int64 index = 0; index < leaf_count; ++index) { in ExecuteOnStream()
33 #define TREE_SIZE(leaf_count) (2 * (leaf_count)-2) argument
110 flattened_list.reserve(sub_shardings.leaf_count()); in Tuple()142 int64 leaf_count = RequiredLeaves(tuple_shape); in SingleTuple() local144 flattened_list.resize(leaf_count, sharding); in SingleTuple()312 const int64 leaf_count = ShapeUtil::GetLeafCount(shape); in RequiredLeaves() local313 return (leaf_count == 0) ? 1 : leaf_count; in RequiredLeaves()
284 TF_RET_CHECK(user_sharding_tree.leaf_count() == 1) in ApplyShardingFromUsers()304 TF_RET_CHECK(sharding_tree.leaf_count() == 1); in ApplyShardingFromUsers()
2619 int leaf_count = ShapeUtil::GetLeafCount(instruction->shape()); in ParseInstructionRhs() local2622 if (leaf_count != replicated.size()) { in ParseInstructionRhs()2624 StrCat("parameter has ", leaf_count, in ParseInstructionRhs()
195 int leaf_count = 0; in Unflatten() local205 leaf_count)); in Unflatten()209 ++leaf_count; in Unflatten()
122 EXPECT_EQ(0, shape_tree.leaf_count()); in TEST_F()
238 int64 leaf_count() const { return std::distance(leaf_begin(), leaf_end()); } in leaf_count() function
557 __u32 leaf_count; member
1299 leaf_node_count = blkid_be32(bnode->leaf_count); in probe_hfsplus()