Searched refs:in_rank (Results 1 – 2 of 2) sorted by relevance
853 void ReduceImpl(OpKernelContext* ctx, OUT_T out, IN_T in, int in_rank,861 } else if (in_rank == 2 && out_rank == 1 &&864 } else if (in_rank == 2 && out_rank == 1 &&867 } else if (in_rank == 3 && out_rank == 2 && reduction_axes[0] == 1) {870 } else if (in_rank == 3 && out_rank == 1 && reduction_axes[0] == 0 &&876 ss << "Invalid reduction requested: in_rank, out_rank, axes " << in_rank
488 int in_rank = c->Rank(in); in __anondb9326b20702() local489 for (int i = 0; i < in_rank; ++i) { in __anondb9326b20702()490 auto in_dim = c->Dim(in, in_rank - i - 1); in __anondb9326b20702()