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