Function GetShape

tensorflow/core/kernels/shape_ops.h:32–36

namespace shape_op_helpers {
inline Status GetShape(OpKernelContext* ctx, int input_index,
                       TensorShape* shape) {
  *shape = ctx->input(input_index).shape();
  return Status::OK();
}
}  // namespace shape_op_helpers

template <typename OutType>