Searched refs:grad_x_reduce_idx (Results 1 – 5 of 5) sorted by relevance
107 Output(ctx, 0, bcast.grad_x_reduce_idx()); in Compile()
100 Output(ctx, 0, bcast.grad_x_reduce_idx()); in Compute()
39 strings::StrAppend(&ret, "[", absl::StrJoin(b.grad_x_reduce_idx(), ","), "]"); in BCast()
380 const Vec& grad_x_reduce_idx() const { return grad_reduce_idx_[0]; } in grad_x_reduce_idx() function
677 reduce_dims[0] = bcast.grad_x_reduce_idx(); in MaterializeBroadcastGradientArgs()