Searched refs:grad_y_reduce_idx (Results 1 – 5 of 5) sorted by relevance
102 const Vec& grad_y_reduce_idx() const { return grad_y_reduce_idx_; } in grad_y_reduce_idx() function
41 strings::StrAppend(&ret, "[", str_util::Join(b.grad_y_reduce_idx(), ","), in BCast()
108 Output(ctx, 1, bcast.grad_y_reduce_idx()); in Compile()
101 Output(ctx, 1, bcast.grad_y_reduce_idx()); in Compute()
624 reduce_dims[1] = bcast.grad_y_reduce_idx(); in MaterializeBroadcastGradientArgs()