Searched refs:in_rank (Results 1 – 2 of 2) sorted by relevance
1073 void ReduceImpl(OpKernelContext* ctx, OUT_T out, IN_T in, int in_rank,1081 } else if (in_rank == 2 && out_rank == 1 &&1084 } else if (in_rank == 2 && out_rank == 1 &&1087 } else if (in_rank == 3 && out_rank == 2 && reduction_axes[0] == 1) {1096 } else if (in_rank == 3 && out_rank == 1 && reduction_axes[0] == 0 &&1102 ss << "Invalid reduction requested: in_rank, out_rank, axes " << in_rank
494 int in_rank = c->Rank(in); in __anondb9326b20702() local495 for (int i = 0; i < in_rank; ++i) { in __anondb9326b20702()496 auto in_dim = c->Dim(in, in_rank - i - 1); in __anondb9326b20702()