Searched refs:grad_y_reduce_idx (Results 1 – 5 of 5) sorted by relevance
108 Output(ctx, 1, bcast.grad_y_reduce_idx()); in Compile()
101 Output(ctx, 1, bcast.grad_y_reduce_idx()); in Compute()
40 strings::StrAppend(&ret, "[", absl::StrJoin(b.grad_y_reduce_idx(), ","), "]"); in BCast()
381 const Vec& grad_y_reduce_idx() const { return grad_reduce_idx_[1]; } in grad_y_reduce_idx() function
678 reduce_dims[1] = bcast.grad_y_reduce_idx(); in MaterializeBroadcastGradientArgs()