Home
last modified time | relevance | path

Searched refs:grad_reduce_idx (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dbcast_test.cc72 strings::StrAppend(&ret, "[", absl::StrJoin(b.grad_reduce_idx(0), ","), "]"); in BCastList3()
73 strings::StrAppend(&ret, "[", absl::StrJoin(b.grad_reduce_idx(1), ","), "]"); in BCastList3()
74 strings::StrAppend(&ret, "[", absl::StrJoin(b.grad_reduce_idx(2), ","), "]"); in BCastList3()
Dbcast.h98 const Vec& grad_reduce_idx(int i) const { return grad_reduce_idx_[i]; } in grad_reduce_idx() function