Searched refs:leaf_begin (Results 1 – 3 of 3) sorted by relevance
195 leaf_iterator leaf_begin() { return leaf_iterator(&nodes_, nodes_.begin()); } in leaf_begin() function197 const_leaf_iterator leaf_begin() const { in leaf_begin() function205 return tensorflow::gtl::make_range(leaf_begin(), leaf_end()); in leaves()208 return tensorflow::gtl::make_range(leaf_begin(), leaf_end()); in leaves()215 return reverse_leaf_iterator(leaf_begin()); in leaf_rend()221 return const_reverse_leaf_iterator(leaf_begin()); in leaf_rend()238 int64 leaf_count() const { return std::distance(leaf_begin(), leaf_end()); } in leaf_count()
57 auto output_leaf_it = output_buffers->leaf_begin(); in ExecuteOnStream()
305 instruction->set_sharding(sharding_tree.leaf_begin()->second); in ApplyShardingFromUsers()