Searched refs:revd_shape (Results 1 – 1 of 1) sorted by relevance
38 const TensorShape revd_shape = ctx->InputShape(1); in Compile() local40 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(revd_shape), in Compile()42 revd_shape.DebugString())); in Compile()43 OP_REQUIRES(ctx, revd_shape.num_elements() == x_shape.dims(), in Compile()44 errors::InvalidArgument("axes ", revd_shape.DebugString(), in Compile()48 if (revd_shape.num_elements() == 0) { in Compile()