Searched refs:axes_shape (Results 1 – 2 of 2) sorted by relevance
76 const TensorShape axes_shape = ctx->InputShape(1); in Compile() local78 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(axes_shape), in Compile()80 axes_shape.DebugString())); in Compile()81 OP_REQUIRES(ctx, axes_shape.num_elements() <= x_shape.dims(), in Compile()82 errors::InvalidArgument("axes ", axes_shape.DebugString(), in Compile()87 if (axes_shape.num_elements() == 0) { in Compile()
3196 axes_shape = array_ops.shape(axes) # [2]3204 array_ops.fill(axes_shape, 1)