Searched refs:grad_reduce_idx (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/util/ |
D | bcast_test.cc | 72 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()
|
D | bcast.h | 98 const Vec& grad_reduce_idx(int i) const { return grad_reduce_idx_[i]; } in grad_reduce_idx() function
|